Update visualizer.py

visualizer.py

@@ -42,6 +42,7 @@ 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)
             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 ] )
@@ -51,6 +52,7 @@ 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