Spaces:
Runtime error
Runtime error
oskarastrom
commited on
Commit
•
5e018ad
1
Parent(s):
f879e7b
Codec checking
Browse files- app.py +3 -9
- temp.mp4 +0 -0
- visualizer.py +9 -0
app.py
CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
|
|
2 |
from uploader import save_data, create_data_dir
|
3 |
from main import predict_task
|
4 |
from state_handler import load_example_result, reset_state
|
|
|
5 |
from file_reader import File
|
6 |
import numpy as np
|
7 |
from aws_handler import upload_file
|
@@ -176,15 +177,8 @@ def show_data():
|
|
176 |
|
177 |
|
178 |
def testing():
|
179 |
-
|
180 |
-
|
181 |
-
print(vr)
|
182 |
-
print("hi")
|
183 |
-
codec = cv2.VideoWriter_fourcc(*'mp4v')
|
184 |
-
vr = cv2.VideoWriter("static/test.mp4", codec, 10, [ int(1.5*100), 100 ] )
|
185 |
-
print(vr)
|
186 |
-
print("hi")
|
187 |
-
vr = cv2.VideoWriter("static/test.mp4", -1, 10, [ int(1.5*100), 100 ] )
|
188 |
testing()
|
189 |
|
190 |
def preview_result(zip_info, aris_info):
|
|
|
2 |
from uploader import save_data, create_data_dir
|
3 |
from main import predict_task
|
4 |
from state_handler import load_example_result, reset_state
|
5 |
+
from visualizer import is_fourcc_available
|
6 |
from file_reader import File
|
7 |
import numpy as np
|
8 |
from aws_handler import upload_file
|
|
|
177 |
|
178 |
|
179 |
def testing():
|
180 |
+
print("avc1", is_fourcc_available("avc1"))
|
181 |
+
print("mp4v", is_fourcc_available("mp4v"))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
182 |
testing()
|
183 |
|
184 |
def preview_result(zip_info, aris_info):
|
temp.mp4
ADDED
Binary file (258 Bytes). View file
|
|
visualizer.py
CHANGED
@@ -20,6 +20,15 @@ BOX_THICKNESS = 2
|
|
20 |
FONT_SCALE = 0.65
|
21 |
FONT_THICKNESS = 1
|
22 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
23 |
def generate_video_batches(didson, preds, frame_rate, video_out_path, gp=None, image_meter_width=None, image_meter_height=None, batch_size=1000):
|
24 |
"""Write a visualized video to video_out_path, given a didson object.
|
25 |
"""
|
|
|
20 |
FONT_SCALE = 0.65
|
21 |
FONT_THICKNESS = 1
|
22 |
|
23 |
+
|
24 |
+
def is_fourcc_available(codec):
|
25 |
+
try:
|
26 |
+
fourcc = cv2.VideoWriter_fourcc(*codec)
|
27 |
+
temp_video = cv2.VideoWriter('temp.mp4', fourcc, 30, (640, 480), isColor=True)
|
28 |
+
return temp_video.isOpened()
|
29 |
+
except:
|
30 |
+
return False
|
31 |
+
|
32 |
def generate_video_batches(didson, preds, frame_rate, video_out_path, gp=None, image_meter_width=None, image_meter_height=None, batch_size=1000):
|
33 |
"""Write a visualized video to video_out_path, given a didson object.
|
34 |
"""
|