freealise commited on
Commit
7c94a8a
·
verified ·
1 Parent(s): 5490605

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -225,23 +225,23 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
225
  blur_frame = blur_image(raw_frame, depth_color, blur_data)
226
  comb_frame = np.concatenate((blur_frame, depth_color), axis=0)
227
 
228
- cv2.imwrite(f"f{count}_comb.jpg", comb_frame, [int(cv2.IMWRITE_JPEG_QUALITY), 100])
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")
233
 
234
- cv2.imwrite(f"f{count}_.jpg", blur_frame)
235
- backup_frames.append(f"f{count}_.jpg")
236
 
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
 
243
- cv2.imwrite(f"f{count}_mask.jpg", depth_gray)
244
- masks.append(f"f{count}_mask.jpg")
245
  count += 1
246
 
247
  final_vid = create_video(comb_frames, frame_rate, "orig")
 
225
  blur_frame = blur_image(raw_frame, depth_color, blur_data)
226
  comb_frame = np.concatenate((blur_frame, depth_color), axis=0)
227
 
228
+ cv2.imwrite(f"f{count}_comb.png", comb_frame)
229
+ comb_frames.append(f"f{count}_comb.png")
230
 
231
+ cv2.imwrite(f"f{count}.png", blur_frame)
232
+ orig_frames.append(f"f{count}.png")
233
 
234
+ cv2.imwrite(f"f{count}_.png", blur_frame)
235
+ backup_frames.append(f"f{count}_.png")
236
 
237
+ cv2.imwrite(f"f{count}_dmap.png", depth_color)
238
+ depth_frames.append(f"f{count}_dmap.png")
239
 
240
  depth_gray = seg_frame(depth_gray, o, b) + 128
241
  #print(depth_gray[depth_gray>128]-128)
242
 
243
+ cv2.imwrite(f"f{count}_mask.png", depth_gray)
244
+ masks.append(f"f{count}_mask.png")
245
  count += 1
246
 
247
  final_vid = create_video(comb_frames, frame_rate, "orig")