oskarastrom commited on
Commit
1edac8d
1 Parent(s): 6041c41

Update visualizer.py

Browse files
Files changed (1) hide show
  1. visualizer.py +1 -1
visualizer.py CHANGED
@@ -34,7 +34,7 @@ def generate_video_batches(didson, preds_path, frame_rate, video_out_path, gp=No
34
  vid_frames, h, w = get_video_frames(frames, preds_path, frame_rate, image_meter_width, image_meter_height, start_frame=i)
35
 
36
  if out is None:
37
- out = cv2.VideoWriter(video_out_path, cv2.VideoWriter_fourcc(*'avc1'), frame_rate, [ int(1.5*w), h ] )
38
 
39
  for j, frame in enumerate(vid_frames):
40
  if gp: gp(( (i+j) / end_frame), 'Generating results video...')
 
34
  vid_frames, h, w = get_video_frames(frames, preds_path, frame_rate, image_meter_width, image_meter_height, start_frame=i)
35
 
36
  if out is None:
37
+ out = cv2.VideoWriter(video_out_path, cv2.VideoWriter_fourcc(*'mp4v'), frame_rate, [ int(1.5*w), h ] )
38
 
39
  for j, frame in enumerate(vid_frames):
40
  if gp: gp(( (i+j) / end_frame), 'Generating results video...')