|
23 | 23 | from src.depthmap_generation import ModelHolder
|
24 | 24 | from src import backbone
|
25 | 25 |
|
26 |
| -# 3d-photo-inpainting imports |
27 |
| -from inpaint.mesh import write_mesh, read_mesh, output_3d_photo |
28 |
| -from inpaint.networks import Inpaint_Color_Net, Inpaint_Depth_Net, Inpaint_Edge_Net |
29 |
| -from inpaint.utils import path_planning |
30 |
| -from inpaint.bilateral_filtering import sparse_bilateral_filtering |
31 |
| - |
32 | 26 | global video_mesh_data, video_mesh_fn
|
33 | 27 | video_mesh_data = None
|
34 | 28 | video_mesh_fn = None
|
@@ -357,6 +351,10 @@ def get_uniquefn(outpath, basename, ext, suffix=''):
|
357 | 351 |
|
358 | 352 |
|
359 | 353 | def run_3dphoto(device, img_rgb, img_depth, inputnames, outpath, gen_inpainted_mesh_demos, vid_ssaa, vid_format):
|
| 354 | + from inpaint.mesh import write_mesh |
| 355 | + from inpaint.networks import Inpaint_Color_Net, Inpaint_Depth_Net, Inpaint_Edge_Net |
| 356 | + from inpaint.bilateral_filtering import sparse_bilateral_filtering |
| 357 | + |
360 | 358 | mesh_fi = ''
|
361 | 359 | try:
|
362 | 360 | print("Running 3D Photo Inpainting .. ")
|
@@ -506,6 +504,8 @@ def run_3dphoto(device, img_rgb, img_depth, inputnames, outpath, gen_inpainted_m
|
506 | 504 |
|
507 | 505 | def run_3dphoto_videos(mesh_fi, basename, outpath, num_frames, fps, crop_border, traj_types, x_shift_range,
|
508 | 506 | y_shift_range, z_shift_range, video_postfix, vid_dolly, vid_format, vid_ssaa):
|
| 507 | + from inpaint.mesh import read_mesh, output_3d_photo |
| 508 | + from inpaint.utils import path_planning |
509 | 509 | import vispy
|
510 | 510 | try:
|
511 | 511 | if platform.system() == 'Windows':
|
|
0 commit comments