Spaces:
Running
on
L40S
Running
on
L40S
bug fixed
Browse files- refine/mesh_refine.py +1 -1
refine/mesh_refine.py
CHANGED
@@ -443,7 +443,7 @@ def geo_refine_3(vertices_, faces_, rgb_ls, fixed_v=None, fixed_f=None, distract
|
|
443 |
|
444 |
# concatenate fixed_v and fixed_f
|
445 |
if fixed_v is not None and fixed_f is not None:
|
446 |
-
dense_vertices, dense_faces = np.concatenate([dense_vertices, fixed_v.detach().cpu().numpy()], axis=0), np.concatenate([dense_faces, fixed_f.detach().cpu().numpy() + len(
|
447 |
dense_vertices, dense_faces = torch.from_numpy(dense_vertices).cuda(), torch.from_numpy(dense_faces.astype('int32')).cuda()
|
448 |
# reconstruct meshes
|
449 |
meshes = Meshes(verts=[dense_vertices], faces=[dense_faces], textures=pytorch3d.renderer.mesh.textures.TexturesVertex([torch.zeros_like(dense_vertices).float()]))
|
|
|
443 |
|
444 |
# concatenate fixed_v and fixed_f
|
445 |
if fixed_v is not None and fixed_f is not None:
|
446 |
+
dense_vertices, dense_faces = np.concatenate([dense_vertices, fixed_v.detach().cpu().numpy()], axis=0), np.concatenate([dense_faces, fixed_f.detach().cpu().numpy() + len(dense_vertices)], axis=0)
|
447 |
dense_vertices, dense_faces = torch.from_numpy(dense_vertices).cuda(), torch.from_numpy(dense_faces.astype('int32')).cuda()
|
448 |
# reconstruct meshes
|
449 |
meshes = Meshes(verts=[dense_vertices], faces=[dense_faces], textures=pytorch3d.renderer.mesh.textures.TexturesVertex([torch.zeros_like(dense_vertices).float()]))
|