Codec checking

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from uploader import save_data, create_data_dir
 from main import predict_task
 from state_handler import load_example_result, reset_state
+from visualizer import is_fourcc_available
 from file_reader import File
 import numpy as np
 from aws_handler import upload_file
@@ -176,15 +177,8 @@ def show_data():
 
     
 def testing():
-    codec = cv2.VideoWriter_fourcc(*'avc1')
-    vr = cv2.VideoWriter("static/test.mp4", codec, 10, [ int(1.5*100), 100 ] )
-    print(vr)
-    print("hi")
-    codec = cv2.VideoWriter_fourcc(*'mp4v')
-    vr = cv2.VideoWriter("static/test.mp4", codec, 10, [ int(1.5*100), 100 ] )
-    print(vr)
-    print("hi")
-    vr = cv2.VideoWriter("static/test.mp4", -1, 10, [ int(1.5*100), 100 ] )
+    print("avc1", is_fourcc_available("avc1"))
+    print("mp4v", is_fourcc_available("mp4v"))
 testing()
 
 def preview_result(zip_info, aris_info):

visualizer.py

@@ -20,6 +20,15 @@ BOX_THICKNESS = 2
 FONT_SCALE = 0.65
 FONT_THICKNESS = 1
 
+
+def is_fourcc_available(codec):
+    try:
+        fourcc = cv2.VideoWriter_fourcc(*codec)
+        temp_video = cv2.VideoWriter('temp.mp4', fourcc, 30, (640, 480), isColor=True)
+        return temp_video.isOpened()
+    except:
+        return False
+
 def generate_video_batches(didson, preds, frame_rate, video_out_path, gp=None, image_meter_width=None, image_meter_height=None, batch_size=1000):
     """Write a visualized video to video_out_path, given a didson object.
     """