Reversed tracking

inference.py

@@ -330,7 +330,7 @@ def do_tracking(all_preds, image_meter_width, image_meter_height, gp=None, max_a
 
     return json_data
 
-def do_associative_tracking(low_preds, high_preds, image_meter_width, image_meter_height, gp=None, low_thresh=0.2, high_threshold=0.2, max_age=MAX_AGE, iou_thres=IOU_THRES, min_hits=MIN_HITS, min_length=MIN_LENGTH, batch_size=BATCH_SIZE, verbose=True):
+def do_associative_tracking(low_preds, high_preds, image_meter_width, image_meter_height, reverse=False, gp=None, max_age=MAX_AGE, iou_thres=IOU_THRES, min_hits=MIN_HITS, min_length=MIN_LENGTH, verbose=True):
 
     if (gp): gp(0, "Tracking...")
 
@@ -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, reversed=True, args={ 'max_age': max_age, 'min_hits': 0, 'iou_threshold': iou_thres}, min_hits=min_hits)
+    tracker = Tracker(clip_info, algorithm=Associate, reverse=reverse, 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(), reverse=True)):
+        for i, key in enumerate(sorted(low_preds.keys(), reverse=reverse)):
             if gp: gp(i / len(low_preds), pbar.__str__())
             low_boxes = low_preds[key]
             high_boxes = high_preds[key]

lib/fish_eye/tracker.py

@@ -9,13 +9,13 @@ 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, reversed=False):
+    def __init__(self, clip_info, algorithm=Sort, args={'max_age':1, 'min_hits':0, 'iou_threshold':0.05}, min_hits=3, reverse=False):
         self.algorithm = algorithm(**args)
         self.fish_ids = Counter()
-        self.reversed = reversed
+        self.reverse = reverse
         self.min_hits = min_hits
         self.json_data = deepcopy(clip_info)
-        if reversed:
+        if reverse:
             self.frame_id = self.json_data['end_frame']
         else:
             self.frame_id = self.json_data['start_frame']
@@ -63,7 +63,7 @@ class Tracker:
                 'frame_num': self.frame_id,
                 'fish': new_frame_entries
             })
-        if self.reversed:
+        if self.reverse:
             self.frame_id -= 1
         else:
             self.frame_id += 1

scripts/detect_frames.py

@@ -31,7 +31,7 @@ def main(args, verbose=True):
     model, device = setup_model(args.weights)
     
     locations = [
-        "kenai-val"
+        "kenai-rightbank"
     ]
     for loc in locations:
 

scripts/track_detection.py

@@ -42,7 +42,7 @@ def main(args, config={}, verbose=True):
 
     
     locations = [
-        "kenai-val"
+        "kenai-rightbank"
     ]
     for loc in locations:
 
@@ -109,7 +109,7 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
         high_outputs = do_suppression(inference, conf_thres=config['high_conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
         high_preds, real_width, real_height = format_predictions(image_shapes, high_outputs, width, height, verbose=verbose)
 
-        results = do_associative_tracking(low_preds, high_preds, image_meter_width, image_meter_height, min_length=config['min_length'], max_age=config['max_age'], iou_thres=config['iou_threshold'], min_hits=config['min_hits'], verbose=verbose)
+        results = do_associative_tracking(low_preds, high_preds, image_meter_width, image_meter_height, reverse=False, min_length=config['min_length'], max_age=config['max_age'], iou_thres=config['iou_threshold'], min_hits=config['min_hits'], verbose=verbose)
     else:
         outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)