Update visualizer.py

visualizer.py

@@ -76,6 +76,7 @@ def get_video_frames(frames, preds, frame_rate, image_meter_width=None, image_me
     preds['frames'] = preds['frames'][start_frame:]
     
     vid_frames = []
+    print(len(frames))
     if len(frames):
         # assumes all frames the same size
         h, w = frames[0].shape
@@ -87,6 +88,8 @@ 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]
@@ -117,7 +120,8 @@ def get_video_frames(frames, preds, frame_rate, image_meter_width=None, image_me
             cv2.putText(frame, f'Frame: {start_frame+i}', (BORDER_PAD, h-BORDER_PAD-LINE_HEIGHT*0), cv2.FONT_HERSHEY_SIMPLEX, FONT_SCALE, WHITE, FONT_THICKNESS, cv2.LINE_AA, False)
 
             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"):