freealise commited on
Commit
a3f556c
·
verified ·
1 Parent(s): fdbdfc2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -243,7 +243,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
243
 
244
  j=0
245
  while j<len(orig_frames):
246
- comb_frame = np.concatenate(cv2.imread(orig_frames[j], cv2.IMREAD_UNCHANGED), cv2.imread(depth_frames[j], cv2.IMREAD_UNCHANGED), axis=0)
247
 
248
  cv2.imwrite(f"f{j}_comb.png", comb_frame)
249
  comb_frames.append(f"f{j}_comb.png")
@@ -458,7 +458,7 @@ def draw_mask(o, b, v, d, evt: gr.EventData):
458
  depth[mask>0] = (0,0,0,0)
459
  cv2.imwrite(depths[frame_selected], depth)
460
 
461
- comb_frame = np.concatenate(frame, depth, axis=0)
462
  cv2.imwrite(comb_frames[frame_selected], comb_frame)
463
 
464
  return gr.ImageEditor(value=d)
 
243
 
244
  j=0
245
  while j<len(orig_frames):
246
+ comb_frame = np.concatenate((cv2.imread(orig_frames[j], cv2.IMREAD_UNCHANGED), cv2.imread(depth_frames[j], cv2.IMREAD_UNCHANGED)), axis=0)
247
 
248
  cv2.imwrite(f"f{j}_comb.png", comb_frame)
249
  comb_frames.append(f"f{j}_comb.png")
 
458
  depth[mask>0] = (0,0,0,0)
459
  cv2.imwrite(depths[frame_selected], depth)
460
 
461
+ comb_frame = np.concatenate((frame, depth), axis=0)
462
  cv2.imwrite(comb_frames[frame_selected], comb_frame)
463
 
464
  return gr.ImageEditor(value=d)