Spaces:
Running
on
L40S
Running
on
L40S
update
Browse files- 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()
|