Update app.py
Browse files
app.py
CHANGED
@@ -137,15 +137,17 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
|
|
137 |
prvs = cv2.cvtColor(fr1, cv2.COLOR_RGBA2GRAY)
|
138 |
hsv = np.zeros_like(fr1)
|
139 |
hsv[..., 1] = 255
|
140 |
-
|
141 |
-
|
142 |
-
flow = res
|
143 |
-
else:
|
144 |
-
res = prvs
|
145 |
|
146 |
while(depth_map.isOpened()):
|
147 |
ret, fr2 = depth_map.read()
|
148 |
if ret == False:
|
|
|
|
|
|
|
|
|
|
|
149 |
break
|
150 |
nxt = cv2.cvtColor(fr2, cv2.COLOR_RGBA2GRAY)
|
151 |
if o_flow == True:
|
@@ -156,7 +158,7 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
|
|
156 |
rgb = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGB)
|
157 |
rgb = cv2.cvtColor(rgb, cv2.COLOR_RGBA2GRAY)
|
158 |
else:
|
159 |
-
rgb =
|
160 |
|
161 |
alpha = 1.0/len(result_frames)
|
162 |
beta = (1.0 - alpha)
|
|
|
137 |
prvs = cv2.cvtColor(fr1, cv2.COLOR_RGBA2GRAY)
|
138 |
hsv = np.zeros_like(fr1)
|
139 |
hsv[..., 1] = 255
|
140 |
+
res = np.zeros_like(prvs)
|
141 |
+
flow = res
|
|
|
|
|
|
|
142 |
|
143 |
while(depth_map.isOpened()):
|
144 |
ret, fr2 = depth_map.read()
|
145 |
if ret == False:
|
146 |
+
if o_flow == False:
|
147 |
+
rgb = prvs
|
148 |
+
alpha = 1.0/len(result_frames)
|
149 |
+
beta = (1.0 - alpha)
|
150 |
+
res = cv2.addWeighted(rgb, alpha, res, beta, 0.0, res)
|
151 |
break
|
152 |
nxt = cv2.cvtColor(fr2, cv2.COLOR_RGBA2GRAY)
|
153 |
if o_flow == True:
|
|
|
158 |
rgb = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGB)
|
159 |
rgb = cv2.cvtColor(rgb, cv2.COLOR_RGBA2GRAY)
|
160 |
else:
|
161 |
+
rgb = prvs
|
162 |
|
163 |
alpha = 1.0/len(result_frames)
|
164 |
beta = (1.0 - alpha)
|