Reverse tracking test

inference.py

@@ -342,11 +342,11 @@ def do_associative_tracking(low_preds, high_preds, image_meter_width, image_mete
         'image_meter_width': image_meter_width,
         'image_meter_height': image_meter_height
     }
-    tracker = Tracker(clip_info, algorithm=Associate, args={ 'max_age': max_age, 'min_hits': 0, 'iou_threshold': iou_thres}, min_hits=min_hits)
+    tracker = Tracker(clip_info, algorithm=Associate, reversed=True, args={ 'max_age': max_age, 'min_hits': 0, 'iou_threshold': iou_thres}, min_hits=min_hits)
     
     # Run tracking
     with tqdm(total=len(low_preds), desc="Running tracking", ncols=0, disable=not verbose) as pbar:
-        for i, key in enumerate(sorted(low_preds.keys())):
+        for i, key in enumerate(sorted(low_preds.keys(), reverse=True)):
             if gp: gp(i / len(low_preds), pbar.__str__())
             low_boxes = low_preds[key]
             high_boxes = high_preds[key]

lib/fish_eye/associative.py

@@ -147,6 +147,7 @@ def associate_detections_to_trackers(detections,trackers,iou_threshold = 0.3):
 
     iou_matrix = iou_batch(detections, trackers)
 
+    # find
     if min(iou_matrix.shape) > 0:
         a = (iou_matrix > iou_threshold).astype(np.int32)
         if a.sum(1).max() == 1 and a.sum(0).max() == 1:
@@ -165,15 +166,15 @@ def associate_detections_to_trackers(detections,trackers,iou_threshold = 0.3):
         if(t not in matched_indices[:,1]):
             unmatched_trackers.append(t)
 
-  #filter out matched with low IOU
+    # filter out matched with low IOU
     matches = []
     for m in matched_indices:
-        if(iou_matrix[m[0], m[1]]<iou_threshold):
+        if (iou_matrix[m[0], m[1]] < iou_threshold):
             unmatched_detections.append(m[0])
             unmatched_trackers.append(m[1])
         else:
             matches.append(m.reshape(1,2))
-    if(len(matches)==0):
+    if (len(matches) == 0):
         matches = np.empty((0,2),dtype=int)
     else:
         matches = np.concatenate(matches,axis=0)

lib/fish_eye/tracker.py

@@ -9,12 +9,16 @@ from sort import Sort
 from associative import Associate
 
 class Tracker:
-    def __init__(self, clip_info, algorithm=Sort, args={'max_age':1, 'min_hits':0, 'iou_threshold':0.05}, min_hits=3):
+    def __init__(self, clip_info, algorithm=Sort, args={'max_age':1, 'min_hits':0, 'iou_threshold':0.05}, min_hits=3, reversed=False):
         self.algorithm = algorithm(**args)
         self.fish_ids = Counter()
+        self.reversed = reversed
         self.min_hits = min_hits
         self.json_data = deepcopy(clip_info)
-        self.frame_id = self.json_data['start_frame']
+        if reversed:
+            self.frame_id = self.json_data['end_frame']
+        else:
+            self.frame_id = self.json_data['start_frame']
         self.json_data['frames'] = []
 
     # Boxes should be given in normalized [x1,y1,x2,y2,c]
@@ -59,7 +63,10 @@ class Tracker:
                 'frame_num': self.frame_id,
                 'fish': new_frame_entries
             })
-        self.frame_id += 1
+        if self.reversed:
+            self.frame_id -= 1
+        else:
+            self.frame_id += 1
     
     def finalize(self, output_path=None, min_length=-1.0): # vert_margin=0.0
         json_data = deepcopy(self.json_data)