update

infer_api.py

@@ -554,7 +554,7 @@ def get_distract_mask(generator, color_0, color_1, normal_0=None, normal_1=None,
 # )
 infer_refine_outside_ratio = 0.20
 
-
+@spaces.GPU()
 def infer_refine(meshes, imgs):
     fixed_v, fixed_f, fixed_t = None, None, None
     flow_vert, flow_vector = None, None
@@ -572,7 +572,7 @@ def infer_refine(meshes, imgs):
         mesh_v, mesh_f = mesh.vertices, mesh.faces
 
         if last_colors is None:
-            @spaces.GPU()
+            # @spaces.GPU()
             def get_mask():
                 mv, proj = make_star_cameras_orthographic(8, 1, r=1.2)
                 mv = mv[[4, 3, 2, 0, 6, 5]]
@@ -642,12 +642,6 @@ def infer_refine(meshes, imgs):
         mesh_v = torch.tensor(mesh_v, dtype=torch.float32)
         mesh_f = torch.tensor(mesh_f)
 
-        print("???", mesh_v.device, mesh_f.device)
-        if fixed_v is not None:
-            print('???" fixed_v', fixed_v.shape, fixed_v.device)
-        if fixed_f is not None:
-            print('???" fixed_f', fixed_f.shape, fixed_f.device)
-
         new_mesh, simp_v, simp_f = geo_refine(mesh_v, mesh_f, colors, normals, fixed_v=fixed_v, fixed_f=fixed_f)
 
         # my mesh flow weight by nearest vertexs

refine/mesh_refine.py

@@ -266,7 +266,6 @@ def run_mesh_refine(vertices, faces, pils: List[Image.Image], fixed_v=None, fixe
     return vertices, faces
 
 
-@spaces.GPU()
 def geo_refine(mesh_v, mesh_f, rgb_ls, normal_ls, expansion_weight=0.1, fixed_v=None, fixed_f=None,
                distract_mask=None, distract_bbox=None, thres=3e-6, no_decompose=False):
     print(mesh_v.device, mesh_f.device)