Try to iterate over predictions axis only?

vanvalenlab · Apr 12, 2021 · a9c8539 · a9c8539
1 parent 8e267dd
commit a9c8539
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepcell_tracking/tracking.py b/deepcell_tracking/tracking.py
@@ -646,9 +646,9 @@ def _get_parent(self, frame, cell, predictions):
                 continue
 
             # only interested in the cells/tracks comparison.
-            num_cells = predictions.shape[1] - len(track_ids)
+            # num_cells = predictions.shape[1] - len(track_ids)
 
-            for cell_idx in range(num_cells):
+            for cell_idx in range(predictions.shape[1]):
                 cell_id = self.idx_to_id[(frame, cell_idx)]
 
                 # probability cell is part of the track