Codec switching

app.py

@@ -176,11 +176,6 @@ def show_data():
 
 
     
-def testing():
-    print("avc1", is_fourcc_available("avc1"))
-    print("mp4v", is_fourcc_available("mp4v"))
-testing()
-
 def preview_result(zip_info, aris_info):
     zip_name = zip_info[0]
     print(zip_name)

visualizer.py

@@ -46,7 +46,8 @@ def generate_video_batches(didson, preds, frame_rate, video_out_path, gp=None, i
             vid_frames, h, w = get_video_frames(frames, preds, frame_rate, image_meter_width, image_meter_height, start_frame=i)
 
             if out is None:
-                out = cv2.VideoWriter(video_out_path, cv2.VideoWriter_fourcc(*'mp4v'), frame_rate, [ int(1.5*w), h ] )
+                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 ] )
 
             for j, frame in enumerate(vid_frames):
                 if gp: gp(( (i+j) / end_frame), 'Generating results video...')