freealise commited on
Commit
63cb1cb
1 Parent(s): 09fe3f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -129,14 +129,14 @@ def infer(url_in,interpolation,fps_output):
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
  cv2.calcOpticalFlowFarneback(prvs, nxt, flow, 0.5, 3, 15, 3, 5, 1.2, 0)
141
  alpha = 1.0/len(frames_list)
142
  beta = (1.0 - alpha)
 
129
  depth_map = cv2.VideoCapture(final_vid)
130
 
131
  ret, fr1 = depth_map.read()
132
+ prvs = cv2.cvtColor(fr1, cv2.COLOR_BGR2GRAY)
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_BGR2GRAY)
140
  cv2.calcOpticalFlowFarneback(prvs, nxt, flow, 0.5, 3, 15, 3, 5, 1.2, 0)
141
  alpha = 1.0/len(frames_list)
142
  beta = (1.0 - alpha)