Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -142,13 +142,20 @@ def image_predict(image):
|
|
142 |
|
143 |
def extract_frames(video):
|
144 |
vidcap = cv2.VideoCapture(video)
|
|
|
|
|
|
|
|
|
145 |
while vidcap.isOpened():
|
|
|
|
|
146 |
success, image = vidcap.read()
|
147 |
if success ==False:
|
148 |
break
|
149 |
-
|
|
|
150 |
|
151 |
-
if res
|
152 |
return (res, render)
|
153 |
return ("", None)
|
154 |
|
|
|
142 |
|
143 |
def extract_frames(video):
|
144 |
vidcap = cv2.VideoCapture(video)
|
145 |
+
vidcap = cv2.VideoCapture(video)
|
146 |
+
fps = vidcap.get(cv2.CAP_PROP_FPS)
|
147 |
+
nof = 5
|
148 |
+
frame_no = 0
|
149 |
while vidcap.isOpened():
|
150 |
+
res = ""
|
151 |
+
render = None
|
152 |
success, image = vidcap.read()
|
153 |
if success ==False:
|
154 |
break
|
155 |
+
if (frame_no*nof)%fps==0:
|
156 |
+
res, render = image_predict(image)
|
157 |
|
158 |
+
if len(res) >0:
|
159 |
return (res, render)
|
160 |
return ("", None)
|
161 |
|