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