JiantaoLin
commited on
Commit
Β·
c380726
1
Parent(s):
b0d4f78
new
Browse files
models/ISOMER/scripts/project_mesh.py
CHANGED
|
@@ -103,9 +103,10 @@ class Pix2FacesRenderer:
|
|
| 103 |
pix_to_face = rast_out[..., -1].to(torch.int32) - 1
|
| 104 |
return pix_to_face
|
| 105 |
|
| 106 |
-
pix2faces_renderer = Pix2FacesRenderer()
|
| 107 |
|
|
|
|
| 108 |
def get_visible_faces(meshes: Meshes, cameras: CamerasBase, resolution=1024):
|
|
|
|
| 109 |
# pix_to_face = render_pix2faces_py3d(meshes, cameras, H=resolution, W=resolution)['pix_to_face']
|
| 110 |
pix_to_face = pix2faces_renderer.render_pix2faces_nvdiff(meshes, cameras, H=resolution, W=resolution)
|
| 111 |
|
|
|
|
| 103 |
pix_to_face = rast_out[..., -1].to(torch.int32) - 1
|
| 104 |
return pix_to_face
|
| 105 |
|
|
|
|
| 106 |
|
| 107 |
+
@spaces.GPU
|
| 108 |
def get_visible_faces(meshes: Meshes, cameras: CamerasBase, resolution=1024):
|
| 109 |
+
pix2faces_renderer = Pix2FacesRenderer()
|
| 110 |
# pix_to_face = render_pix2faces_py3d(meshes, cameras, H=resolution, W=resolution)['pix_to_face']
|
| 111 |
pix_to_face = pix2faces_renderer.render_pix2faces_nvdiff(meshes, cameras, H=resolution, W=resolution)
|
| 112 |
|