Update app.py
Browse files
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)
|