rahul5035 commited on
Commit
0ed07d4
·
1 Parent(s): 91ee655

Update prediction.py

Browse files
Files changed (1) hide show
  1. 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(*'vp80'),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=""
 
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=""