YulianSa commited on
Commit
ccc749d
·
1 Parent(s): 402acb5
Files changed (1) hide show
  1. refine/mesh_refine.py +1 -1
refine/mesh_refine.py CHANGED
@@ -327,4 +327,4 @@ def geo_refine_3(vertices, faces, rgb_ls, fixed_v=None, fixed_f=None, distract_m
327
  if fixed_v is not None and fixed_f is not None:
328
  new_meshes = Meshes(verts=[new_meshes.verts_packed()[:origin_len_v]], faces=[new_meshes.faces_packed()[:origin_len_f]],
329
  textures=pytorch3d.renderer.mesh.textures.TexturesVertex([new_meshes.textures.verts_features_packed()[:origin_len_v]]))
330
- return new_meshes.to("cpu"), vertices, faces
 
327
  if fixed_v is not None and fixed_f is not None:
328
  new_meshes = Meshes(verts=[new_meshes.verts_packed()[:origin_len_v]], faces=[new_meshes.faces_packed()[:origin_len_f]],
329
  textures=pytorch3d.renderer.mesh.textures.TexturesVertex([new_meshes.textures.verts_features_packed()[:origin_len_v]]))
330
+ return new_meshes.to("cpu"), vertices.cpu(), faces.cpu()