Fix coord bug in centroid_aug.py: optimize centroid box calculatio

xing-lab-pitt · May 29, 2024 · a03937d · a03937d
1 parent 479506f
commit a03937d
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/notebooks/scripts/mmdetection_classify/centroid_aug.py b/notebooks/scripts/mmdetection_classify/centroid_aug.py
@@ -62,14 +62,24 @@ def __call__(self, results, exact_match_rg_channel=False):
             properties.sort(key=lambda x: x.area, reverse=True)
             props = properties[0]
             centroid = props.centroid
+            # centroid_box = np.array(
+            #     [
+            #         centroid[0] - centroid_box_width / 2,
+            #         centroid[1] - centroid_box_width / 2,
+            #         centroid[0] + centroid_box_width / 2,
+            #         centroid[1] + centroid_box_width / 2,
+            #     ]
+            # )
+
             centroid_box = np.array(
                 [
-                    centroid[0] - centroid_box_width / 2,
                     centroid[1] - centroid_box_width / 2,
-                    centroid[0] + centroid_box_width / 2,
+                    centroid[0] - centroid_box_width / 2,
                     centroid[1] + centroid_box_width / 2,
+                    centroid[0] + centroid_box_width / 2,
                 ]
             )
+
             # Clip to image size
             centroid_box[0] = max(0, centroid_box[0])
             centroid_box[1] = max(0, centroid_box[1])