freealise commited on
Commit
ac1eb99
1 Parent(s): 099e1d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -178,8 +178,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', remove_bg
178
  #
179
  depth = predict_depth(raw_frame[:, :, ::-1], model)
180
  depth = (depth - depth.min()) / (depth.max() - depth.min()) * 255.0
181
- depth = depth.astype(np.uint8)
182
- depth_color = Image.fromarray(depth)
183
  depth_gray = cv2.cvtColor(depth_color, cv2.COLOR_RGBA2GRAY)
184
  #
185
 
 
178
  #
179
  depth = predict_depth(raw_frame[:, :, ::-1], model)
180
  depth = (depth - depth.min()) / (depth.max() - depth.min()) * 255.0
181
+ depth_color = depth.astype(np.uint8)
 
182
  depth_gray = cv2.cvtColor(depth_color, cv2.COLOR_RGBA2GRAY)
183
  #
184