xinjie.wang commited on
Commit
a73ef9b
·
1 Parent(s): d7f41a5
Files changed (1) hide show
  1. asset3d_gen/data/backproject_v2.py +6 -10
asset3d_gen/data/backproject_v2.py CHANGED
@@ -426,19 +426,15 @@ class TextureBacker:
426
  ) -> trimesh.Trimesh:
427
  self._lazy_init_render(self.camera_params, self.mask_thresh)
428
 
429
- self.vertices = torch.from_numpy(mesh.vertices).to(self.device).float()
430
- self.faces = torch.from_numpy(mesh.faces).to(self.device).to(torch.int)
431
- self.uv_map = torch.from_numpy(mesh.visual.uv).to(self.device).float()
432
 
433
- rendered_depth, masks = self.renderer.render_depth(
434
- self.vertices, self.faces
435
- )
436
  norm_deps = self.renderer.normalize_map_by_mask(rendered_depth, masks)
437
- render_uvs, _ = self.renderer.render_uv(
438
- self.vertices, self.faces, self.uv_map
439
- )
440
  view_normals = self.compute_enhanced_viewnormal(
441
- self.renderer.mv_mtx, self.vertices, self.faces
442
  )
443
 
444
  textures, weighted_cos_maps = [], []
 
426
  ) -> trimesh.Trimesh:
427
  self._lazy_init_render(self.camera_params, self.mask_thresh)
428
 
429
+ vertices = torch.from_numpy(mesh.vertices).to(self.device).float()
430
+ faces = torch.from_numpy(mesh.faces).to(self.device).to(torch.int)
431
+ uv_map = torch.from_numpy(mesh.visual.uv).to(self.device).float()
432
 
433
+ rendered_depth, masks = self.renderer.render_depth(vertices, faces)
 
 
434
  norm_deps = self.renderer.normalize_map_by_mask(rendered_depth, masks)
435
+ render_uvs, _ = self.renderer.render_uv(vertices, faces, uv_map)
 
 
436
  view_normals = self.compute_enhanced_viewnormal(
437
+ self.renderer.mv_mtx, vertices, faces
438
  )
439
 
440
  textures, weighted_cos_maps = [], []