Skip to content

Commit 1e92b5f

Browse files
committed
Isolate vispy away
It caused import issues for some users Closes #426
1 parent ee54b55 commit 1e92b5f

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/core.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -23,12 +23,6 @@
2323
from src.depthmap_generation import ModelHolder
2424
from src import backbone
2525

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-
3226
global video_mesh_data, video_mesh_fn
3327
video_mesh_data = None
3428
video_mesh_fn = None
@@ -357,6 +351,10 @@ def get_uniquefn(outpath, basename, ext, suffix=''):
357351

358352

359353
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+
360358
mesh_fi = ''
361359
try:
362360
print("Running 3D Photo Inpainting .. ")
@@ -506,6 +504,8 @@ def run_3dphoto(device, img_rgb, img_depth, inputnames, outpath, gen_inpainted_m
506504

507505
def run_3dphoto_videos(mesh_fi, basename, outpath, num_frames, fps, crop_border, traj_types, x_shift_range,
508506
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
509509
import vispy
510510
try:
511511
if platform.system() == 'Windows':

0 commit comments

Comments
 (0)