Improve debug plots

neuroinformatics-unit · Sep 25, 2024 · 6000609 · 6000609
1 parent 9bf17eb
commit 6000609
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/derotation/analysis/incremental_rotation_pipeline.py b/derotation/analysis/incremental_rotation_pipeline.py
@@ -575,13 +575,21 @@ def plot_center_distribution(
         ax.imshow(self.image_stack[0])
         for center in centers:
             ax.scatter(center[0], center[1], color="red", marker="x")
+        #  plot theoretical center in green
+        ax.scatter(
+            len(self.image_stack[0]) // 2,
+            len(self.image_stack[0]) // 2,
+            color="green",
+            marker="x",
+        )
         ax.scatter(mean_center[0], mean_center[1], color="green", label="Mean")
         ax.scatter(
             median_center[0], median_center[1], color="blue", label="Median"
         )
         ax.scatter(
             mode_center[0], mode_center[1], color="purple", label="Mode"
         )
+        ax.axis("off")
         ax.legend()
 
         plt.savefig(