freealise commited on
Commit
eea5784
·
verified ·
1 Parent(s): 8001546

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -223,6 +223,10 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
223
  thumbnail_old = thumbnail
224
 
225
  blur_frame = blur_image(raw_frame, depth_color, blur_data)
 
 
 
 
226
 
227
  cv2.imwrite(f"f{count}.jpg", blur_frame)
228
  orig_frames.append(f"f{count}.jpg")
@@ -233,9 +237,6 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
233
  cv2.imwrite(f"f{count}_dmap.jpg", depth_color)
234
  depth_frames.append(f"f{count}_dmap.jpg")
235
 
236
- comb_frames.append(f"f{count}.jpg")
237
- comb_frames.append(f"f{count}_dmap.jpg")
238
-
239
  depth_gray = seg_frame(depth_gray, o, b) + 128
240
  #print(depth_gray[depth_gray>128]-128)
241
 
 
223
  thumbnail_old = thumbnail
224
 
225
  blur_frame = blur_image(raw_frame, depth_color, blur_data)
226
+ comb_frame = np.concatenate((blur_frame, depth_color), axis=1)
227
+
228
+ cv2.imwrite(f"f{count}_comb.jpg", comb_frame)
229
+ comb_frames.append(f"f{count}_comb.jpg")
230
 
231
  cv2.imwrite(f"f{count}.jpg", blur_frame)
232
  orig_frames.append(f"f{count}.jpg")
 
237
  cv2.imwrite(f"f{count}_dmap.jpg", depth_color)
238
  depth_frames.append(f"f{count}_dmap.jpg")
239
 
 
 
 
240
  depth_gray = seg_frame(depth_gray, o, b) + 128
241
  #print(depth_gray[depth_gray>128]-128)
242