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]
|
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)
|