Fix: Isolate vispy away

Author: semjonsona

inpaint/utils.py

@@ -972,7 +972,7 @@ def follow_image_aspect_ratio(depth, image):
     return depth
 
 def depth_resize(depth, origin_size, image_size):
-    if origin_size[0] is not 0:
+    if origin_size[0] != 0:
         max_depth = depth.max()
         depth = depth / max_depth
         depth = resize(depth, origin_size, order=1, mode='edge')

src/core.py

@@ -23,6 +23,17 @@
 from src.depthmap_generation import ModelHolder
 from src import backbone
 
+try:
+    # 3d-photo-inpainting imports
+    from inpaint.mesh import write_mesh, read_mesh, output_3d_photo
+    from inpaint.networks import Inpaint_Color_Net, Inpaint_Depth_Net, Inpaint_Edge_Net
+    from inpaint.utils import path_planning
+    from inpaint.bilateral_filtering import sparse_bilateral_filtering
+except Exception as e:
+    print('Impaint import failed. Impaint will not work.')
+    import traceback
+    traceback.print_exc()
+
 global video_mesh_data, video_mesh_fn
 video_mesh_data = None
 video_mesh_fn = None
@@ -84,6 +95,7 @@ def core_generation_funnel(outpath, inputimages, inputdepthmaps, inputnames, inp
 
     # TODO: ideally, run_depthmap should not save meshes - that makes the function not pure
     print(SCRIPT_FULL_NAME)
+    print(f'Backbone: {backbone.USED_BACKBONE.name}')
 
     backbone.unload_sd_model()
 
@@ -351,10 +363,6 @@ def get_uniquefn(outpath, basename, ext, suffix=''):
 
 
 def run_3dphoto(device, img_rgb, img_depth, inputnames, outpath, gen_inpainted_mesh_demos, vid_ssaa, vid_format):
-    from inpaint.mesh import write_mesh
-    from inpaint.networks import Inpaint_Color_Net, Inpaint_Depth_Net, Inpaint_Edge_Net
-    from inpaint.bilateral_filtering import sparse_bilateral_filtering
-
     mesh_fi = ''
     try:
         print("Running 3D Photo Inpainting .. ")
@@ -504,8 +512,6 @@ def run_3dphoto(device, img_rgb, img_depth, inputnames, outpath, gen_inpainted_m
 
 def run_3dphoto_videos(mesh_fi, basename, outpath, num_frames, fps, crop_border, traj_types, x_shift_range,
                        y_shift_range, z_shift_range, video_postfix, vid_dolly, vid_format, vid_ssaa):
-    from inpaint.mesh import read_mesh, output_3d_photo
-    from inpaint.utils import path_planning
     import vispy
     try:
         if platform.system() == 'Windows':