freealise commited on
Commit
74ba0d7
·
verified ·
1 Parent(s): 706d65e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -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")
@@ -478,13 +478,14 @@ def get_mean_and_std(x):
478
  return x_mean, x_std
479
 
480
  def color_transfer(base):
 
481
  global frames
482
  targets = []
483
 
484
  for n in range(len(frames)):
 
485
  targets.append(frames[base])
486
  if n != base:
487
- print("Converting picture "+str(n)+"...")
488
  s, t = read_file(frames[n],targets[n])
489
  s_mean, s_std = get_mean_and_std(s)
490
  t_mean, t_std = get_mean_and_std(t)
 
243
 
244
  j=0
245
  while j<len(orig_frames):
246
+ comb_frame = np.concatenate((cv2.imread(orig_frames[j]), cv2.imread(depth_frames[j])), axis=0)
247
 
248
  cv2.imwrite(f"f{j}_comb.png", comb_frame)
249
  comb_frames.append(f"f{j}_comb.png")
 
478
  return x_mean, x_std
479
 
480
  def color_transfer(base):
481
+ print("Color transfer:")
482
  global frames
483
  targets = []
484
 
485
  for n in range(len(frames)):
486
+ print("Converting picture "+str(n)+"...")
487
  targets.append(frames[base])
488
  if n != base:
 
489
  s, t = read_file(frames[n],targets[n])
490
  s_mean, s_std = get_mean_and_std(s)
491
  t_mean, t_std = get_mean_and_std(t)