Compare commits

..

No commits in common. "viewfix" and "main" have entirely different histories.

View File

@ -64,7 +64,7 @@ def view_annotations(
annotations = recording.annotations annotations = recording.annotations
# 2. Setup Color Mapping # 2. Setup Color Mapping
palette = ["#2196F3", "#9C27B0", "#EAF4FE", "#7B1FA2", "#5C6BC0", "#CE93D8", "#1565C0", "#7C4DFF"] palette = ["#2196F3", "#9C27B0", "#64B5F6", "#7B1FA2", "#5C6BC0", "#CE93D8", "#1565C0", "#7C4DFF"]
unique_labels = sorted(list(set(ann.label for ann in annotations if ann.label))) unique_labels = sorted(list(set(ann.label for ann in annotations if ann.label)))
label_to_color = {label: palette[i % len(palette)] for i, label in enumerate(unique_labels)} label_to_color = {label: palette[i % len(palette)] for i, label in enumerate(unique_labels)}
@ -102,14 +102,13 @@ def view_annotations(
] ]
ax.legend(handles=legend_elements, loc="upper right", framealpha=0.2) ax.legend(handles=legend_elements, loc="upper right", framealpha=0.2)
ax.set_title(title, fontsize=title_fontsize, pad=14) ax.set_title(title, fontsize=title_fontsize, pad=20)
ax.set_xlabel("Time (s)", fontsize=12) ax.set_xlabel("Time (s)", fontsize=12)
ax.set_ylabel("Frequency (MHz)", fontsize=12) ax.set_ylabel("Frequency (MHz)", fontsize=12)
ax.yaxis.set_major_formatter(ticker.FuncFormatter(lambda y, _: f"{y / 1e6:.1f}"))
ax.grid(alpha=0.1) ax.grid(alpha=0.1)
output_path, _ = set_path(output_path=output_path) output_path, _ = set_path(output_path=output_path)
plt.savefig(output_path, dpi=dpi, bbox_inches="tight", pad_inches=0.2) plt.savefig(output_path, dpi=dpi, bbox_inches="tight")
plt.close(fig) plt.close(fig)
print(f"Professional annotation plot saved to {output_path}") print(f"Professional annotation plot saved to {output_path}")