Merge pull request #257 from computational-cell-analytics/m-fix-bb

Fix Bounding Box Sorting
computational-cell-analytics · Nov 1, 2023 · d7a4365 · d7a4365
2 parents 02199d1 + 0d5f967
commit d7a4365
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/micro_sam/training/util.py b/micro_sam/training/util.py
@@ -86,8 +86,8 @@ def _get_prompt_lists(self, gt, n_samples, prompt_generator):
             sampled_cell_ids = np.random.choice(cell_ids, size=min(n_samples, len(cell_ids)), replace=False)
             sampled_cell_ids = np.sort(sampled_cell_ids)
 
-            # only keep the bounding boxes for sampled cell ids
-            bbox_coordinates = [bbox_coordinates[sampled_id] for sampled_id in sampled_cell_ids]
+        # only keep the bounding boxes for sampled cell ids
+        bbox_coordinates = [bbox_coordinates[sampled_id] for sampled_id in sampled_cell_ids]
 
         # convert the gt to the one-hot-encoded masks for the sampled cell ids
         object_masks = segmentation_to_one_hot(gt, None if n_samples is None else sampled_cell_ids)