2lu commited on
Commit
c23f366
·
verified ·
1 Parent(s): 70e3686

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
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
- res, render = image_predict(image)
 
150
 
151
- if res == "car car accident detected":
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