YulianSa commited on
Commit
522a413
·
1 Parent(s): d8f0376
Files changed (1) hide show
  1. 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(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()]))
 
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()]))