freealise commited on
Commit
cbfad45
1 Parent(s): 0dd8126

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -128,15 +128,15 @@ def infer(url_in,interpolation,fps_output):
128
 
129
  depth_map = cv2.VideoCapture(final_vid)
130
 
131
- ret, frame1 = depth_map.read()
132
- prvs = cv2.cvtColor(frame1, cv2.COLOR_RGB2GRAY)
133
  res = prvs
134
  flow = res
135
  while(depth_map.isOpened()):
136
- ret, frame2 = depth_map.read()
137
  if ret == False:
138
  break
139
- nxt = cv2.cvtColor(frame2, cv2.COLOR_RGB2GRAY)
140
  flow = cv2.calcOpticalFlowFarneback(prvs, nxt, None, 0.5, 3, 15, 3, 5, 1.2, 0)
141
  alpha = 1.0/len(frames_list)
142
  beta = (1.0 - alpha)
 
128
 
129
  depth_map = cv2.VideoCapture(final_vid)
130
 
131
+ ret, fr1 = depth_map.read()
132
+ prvs = cv2.cvtColor(fr1, cv2.COLOR_RGB2GRAY)
133
  res = prvs
134
  flow = res
135
  while(depth_map.isOpened()):
136
+ ret, fr2 = depth_map.read()
137
  if ret == False:
138
  break
139
+ nxt = cv2.cvtColor(fr2, cv2.COLOR_RGB2GRAY)
140
  flow = cv2.calcOpticalFlowFarneback(prvs, nxt, None, 0.5, 3, 15, 3, 5, 1.2, 0)
141
  alpha = 1.0/len(frames_list)
142
  beta = (1.0 - alpha)