Spaces:
Running
on
Zero
Running
on
Zero
xinjie.wang
commited on
Commit
·
a73ef9b
1
Parent(s):
d7f41a5
update
Browse files
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 |
-
|
430 |
-
|
431 |
-
|
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,
|
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 = [], []
|