Testing tracker conf

inference.py

@@ -365,6 +365,7 @@ 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
     }
+    print("Tracking using Associate")
     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

lib/fish_eye/tracker.py

@@ -29,6 +29,7 @@ class Tracker:
             # Match confidence with correct track
             conf = 0
             min_score = 1000000
+            print("algorithm", self.algorithm)
             if self.algorithm == Sort:
                 print("SORT")
                 for det in dets:

visualizer.py

@@ -42,9 +42,6 @@ def generate_video_batches(didson, preds, frame_rate, video_out_path, gp=None, i
             frames = didson.load_frames(start_frame=i, end_frame=batch_end)
             vid_frames, h, w = get_video_frames(frames, preds, frame_rate, image_meter_width, image_meter_height, start_frame=i)
 
-            print("starting", i)
-            print("len frames", len(frames))
-            print("len vid_frames", len(vid_frames))
             if out is None:
                 codec = cv2.VideoWriter_fourcc(*'avc1') if is_fourcc_available("avc1") else cv2.VideoWriter_fourcc(*'mp4v')
                 out = cv2.VideoWriter(video_out_path, codec, frame_rate, [ int(1.5*w), h ] )
@@ -54,7 +51,6 @@ def generate_video_batches(didson, preds, frame_rate, video_out_path, gp=None, i
                 out.write(frame)
                 pbar.update(1)
 
-            print("done", i)
             del frames
             del vid_frames
     
@@ -74,11 +70,8 @@ def get_video_frames(frames, preds, frame_rate, image_meter_width=None, image_me
     
     # filter JSON, if necessary (for shorter clips)
     preds_frames = preds['frames'][start_frame:]
-    print(len(preds['frames']))
-    print(start_frame)
     
     vid_frames = []
-    print(len(frames))
     if len(frames):
         # assumes all frames the same size
         h, w = frames[0].shape
@@ -90,8 +83,6 @@ def get_video_frames(frames, preds, frame_rate, image_meter_width=None, image_me
 
         num_frames = min(len(frames), len(preds_frames))
         
-        print(len(frames[:num_frames]))
-        print(num_frames)
         for i, frame_raw in enumerate(frames[:num_frames]):
             frame_raw = cv2.resize(cv2.cvtColor(frame_raw, cv2.COLOR_GRAY2BGR), (w,h))
             pred = preds_frames[i]
@@ -123,7 +114,6 @@ def get_video_frames(frames, preds, frame_rate, image_meter_width=None, image_me
 
             vid_frames.append(frame)
     
-    print(len(vid_frames))
     return vid_frames, h, w
 
 def draw_fish(frame, left, right, top, bottom, color, fish_id, fish_len, LINE_HEIGHT=18, anno_align="left"):