freealise commited on
Commit
6832791
1 Parent(s): 66a772a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -131,7 +131,7 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize):
131
  files = final_vid
132
 
133
  depth_map = cv2.VideoCapture(final_vid)
134
- print("total frames: " + str(len(result_frames)))
135
 
136
  ret, fr1 = depth_map.read()
137
  prvs = cv2.cvtColor(fr1, cv2.COLOR_RGBA2GRAY)
@@ -150,7 +150,7 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize):
150
  hsv[..., 2] = cv2.normalize(mag, None, 0, 255, cv2.NORM_MINMAX)
151
  rgb = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGB)
152
  rgb = cv2.cvtColor(rgb, cv2.COLOR_RGBA2GRAY)
153
- alpha = 1.0/len(result_frames)
154
  beta = (1.0 - alpha)
155
  res = cv2.addWeighted(rgb, alpha, res, beta, 0.0, res)
156
  prvs = nxt
 
131
  files = final_vid
132
 
133
  depth_map = cv2.VideoCapture(final_vid)
134
+ print("interpolated frames: " + str(len(frames_list)) + " -> " + str(len(result_frames)))
135
 
136
  ret, fr1 = depth_map.read()
137
  prvs = cv2.cvtColor(fr1, cv2.COLOR_RGBA2GRAY)
 
150
  hsv[..., 2] = cv2.normalize(mag, None, 0, 255, cv2.NORM_MINMAX)
151
  rgb = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGB)
152
  rgb = cv2.cvtColor(rgb, cv2.COLOR_RGBA2GRAY)
153
+ alpha = 1.0/len(frames_list)
154
  beta = (1.0 - alpha)
155
  res = cv2.addWeighted(rgb, alpha, res, beta, 0.0, res)
156
  prvs = nxt