423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-23 06:12:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

Mesh normal consistency when no faces intersect

Browse Source 
Summary: Corner case where there's nothing to do in this function.

Reviewed By: nikhilaravi

Differential Revision: D26073476

fbshipit-source-id: eb061683ffe35c1ffa8384c422a1557a636d52cd
This commit is contained in:
Jeremy Reizenstein 2021-01-27 17:30:25 -08:00 committed by Facebook GitHub Bot
parent 7f62eacdb2
commit e42b0c4f70
2 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pytorch3d/loss/mesh_normal_consistency.py
View File

@ -114,6 +114,11 @@ def mesh_normal_consistency(meshes):
vert_edge_pair_idx, device=meshes.device, dtype=torch.int64
)
if vert_edge_pair_idx.shape[0] == 0:
return torch.tensor(
[0.0], dtype=torch.float32, device=meshes.device, requires_grad=True
)
v0_idx = edges_packed[edge_idx, 0]
v0 = verts_packed[v0_idx]
v1_idx = edges_packed[edge_idx, 1]

11
tests/test_mesh_normal_consistency.py
View File

@ -218,6 +218,17 @@ class TestMeshNormalConsistency(unittest.TestCase):
self.assertTrue(torch.allclose(out1, out2))
def test_no_intersection(self):
"""
Test Mesh Normal Consistency for a mesh known to have no
intersecting faces.
"""
verts = torch.rand(1, 6, 2)
faces = torch.arange(6).reshape(1, 2, 3)
meshes = Meshes(verts=verts, faces=faces)
out = mesh_normal_consistency(meshes)
self.assertEqual(out.item(), 0)
@staticmethod
def mesh_normal_consistency_with_ico(
num_meshes: int, level: int = 3, device: str = "cpu"