Spaces:
Running
Running
Update prediction.py
Browse files- prediction.py +1 -1
prediction.py
CHANGED
@@ -23,7 +23,7 @@ class Predict:
|
|
23 |
original_video_width=int(video_reader.get(cv2.CAP_PROP_FRAME_WIDTH))
|
24 |
original_video_hight=int(video_reader.get(cv2.CAP_PROP_FRAME_HEIGHT))
|
25 |
|
26 |
-
video_writer=cv2.VideoWriter(output_path,cv2.VideoWriter_fourcc(
|
27 |
frames_deque=deque(maxlen=frames_needed)
|
28 |
|
29 |
predicted_class_name=""
|
|
|
23 |
original_video_width=int(video_reader.get(cv2.CAP_PROP_FRAME_WIDTH))
|
24 |
original_video_hight=int(video_reader.get(cv2.CAP_PROP_FRAME_HEIGHT))
|
25 |
|
26 |
+
video_writer=cv2.VideoWriter(output_path,cv2.VideoWriter_fourcc('V', 'P', '9', '0'),video_reader.get(cv2.CAP_PROP_FPS),(original_video_width,original_video_hight))
|
27 |
frames_deque=deque(maxlen=frames_needed)
|
28 |
|
29 |
predicted_class_name=""
|