Update app.py
Browse files
app.py
CHANGED
@@ -158,15 +158,15 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
|
|
158 |
hsv[..., 0] = ang*180/np.pi/2
|
159 |
hsv[..., 2] = cv2.normalize(mag, None, 0, 255, cv2.NORM_MINMAX)
|
160 |
rgb = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGB)
|
161 |
-
|
|
|
|
|
162 |
else:
|
163 |
rgb = prvs
|
164 |
|
165 |
alpha = 1.0/len(result_frames)
|
166 |
beta = (1.0 - alpha)
|
167 |
res = cv2.addWeighted(rgb, alpha, res, beta, 0.0, res)
|
168 |
-
cv2.imwrite(f"opticalfb{i}.jpg", rgb)
|
169 |
-
depth_frames.append(f"opticalfb{i}.jpg")
|
170 |
i+=1
|
171 |
prvs = nxt
|
172 |
|
|
|
158 |
hsv[..., 0] = ang*180/np.pi/2
|
159 |
hsv[..., 2] = cv2.normalize(mag, None, 0, 255, cv2.NORM_MINMAX)
|
160 |
rgb = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGB)
|
161 |
+
cv2.imwrite(f"opticalfb{i}.jpg", rgb)
|
162 |
+
depth_frames.append(f"opticalfb{i}.jpg")
|
163 |
+
rgb = cv2.cvtColor(rgb, cv2.COLOR_RGBA2GRAY)
|
164 |
else:
|
165 |
rgb = prvs
|
166 |
|
167 |
alpha = 1.0/len(result_frames)
|
168 |
beta = (1.0 - alpha)
|
169 |
res = cv2.addWeighted(rgb, alpha, res, beta, 0.0, res)
|
|
|
|
|
170 |
i+=1
|
171 |
prvs = nxt
|
172 |
|