update

infer_api.py

@@ -642,7 +642,13 @@ def infer_refine(meshes, imgs):
         mesh_v = torch.tensor(mesh_v, dtype=torch.float32)
         mesh_f = torch.tensor(mesh_f)
 
-        new_mesh, simp_v, simp_f = geo_refine(mesh_v, mesh_f, colors, normals, fixed_v=fixed_v, fixed_f=fixed_f, distract_mask=distract_mask, distract_bbox=distract_bbox)
+        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
         try:

refine/mesh_refine.py

@@ -266,6 +266,7 @@ 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)
@@ -278,7 +279,6 @@ def geo_refine(mesh_v, mesh_f, rgb_ls, normal_ls, expansion_weight=0.1, fixed_v=
     vertices, faces = geo_refine_2(vertices, faces, fixed_v=fixed_v)
     return geo_refine_3(vertices, faces, rgb_ls, fixed_v=fixed_v, fixed_f=fixed_f, distract_mask=distract_mask)
 
-@spaces.GPU()
 def geo_refine_1(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):
     
@@ -318,7 +318,6 @@ def geo_refine_2(vertices, faces, fixed_v=None):
         vertices, faces = trimesh.remesh.subdivide(vertices, faces)
     return vertices, faces
 
-@spaces.GPU()
 def geo_refine_3(vertices, faces, rgb_ls, fixed_v=None, fixed_f=None, distract_mask=None):
     origin_len_v, origin_len_f = len(vertices), len(faces)
     # concatenate fixed_v and fixed_f