x

scripts/multiview_inference.py

@@ -14,7 +14,6 @@ from pytorch3d.structures import Meshes, join_meshes_as_scene
 import numpy as np
 
 def fast_geo(front_normal: Image.Image, back_normal: Image.Image, side_normal: Image.Image, clamp=0., init_type="std"):
-    import time
     if front_normal.mode == "RGB":
         front_normal = simple_remove(front_normal, run_sr=False)
     front_normal = front_normal.resize((192, 192))
@@ -59,7 +58,7 @@ def erode_alpha(img_list):
         img = Image.fromarray(np.concatenate([arr[:, :, :3], alpha[:, :, None]], axis=-1))
         out_img_list.append(img)
     return out_img_list
-import time
+
 def geo_reconstruct(rgb_pils, normal_pils, front_pil, do_refine=False, predict_normal=True, expansion_weight=0.1, init_type="std"):
     if front_pil.size[0] <= 512:
         front_pil = run_sr_fast([front_pil])[0]