423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 03:42:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

Submesh 1/n: Implement mesh equality

Browse Source 
Summary: Adding a mesh equality operator. Two Meshes objects m1, m2 are equal iff their vertex lists, face lists, and normals lists are equal. Textures meshes are not supported yet, but will be added for vertex textures down the stack.

Reviewed By: bottler, nikhilaravi

Differential Revision: D35440655

fbshipit-source-id: 69974a59c091416afdb2892896859a189f5ebf3a
This commit is contained in:
Krzysztof Chalupka 2022-04-11 16:27:53 -07:00 committed by Facebook GitHub Bot
parent aab95575a6
commit 7f097b064b
1 changed files with 51 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
tests/test_meshes.py
View File

@ -127,6 +127,45 @@ def init_simple_mesh(device: str = "cpu"):
return Meshes(verts=verts, faces=faces)
def mesh_structures_equal(mesh1, mesh2) -> bool:
"""
Two meshes are equal if they have identical verts_list and faces_list.
Use to_sorted() before passing into this function to obtain meshes invariant to
vertex permutations. Note that this operator treats two geometrically identical
meshes as different if their vertices are in different coordinate frames.
"""
if mesh1.__class__ != mesh1.__class__:
return False
if mesh1.textures is not None or mesh2.textures is not None:
raise NotImplementedError(
"mesh equality is not implemented for textured meshes."
)
if len(mesh1.verts_list()) != len(mesh2.verts_list()) or not all(
torch.equal(verts_mesh1, verts_mesh2)
for (verts_mesh1, verts_mesh2) in zip(mesh1.verts_list(), mesh2.verts_list())
):
return False
if len(mesh1.faces_list()) != len(mesh2.faces_list()) or not all(
torch.equal(faces_mesh1, faces_mesh2)
for (faces_mesh1, faces_mesh2) in zip(mesh1.faces_list(), mesh2.faces_list())
):
return False
if len(mesh1.verts_normals_list()) != len(mesh2.verts_normals_list()) or not all(
torch.equal(normals_mesh1, normals_mesh2)
for (normals_mesh1, normals_mesh2) in zip(
mesh1.verts_normals_list(), mesh2.verts_normals_list()
)
):
return False
return True
class TestMeshes(TestCaseMixin, unittest.TestCase):
def setUp(self) -> None:
np.random.seed(42)
@ -1172,6 +1211,18 @@ class TestMeshes(TestCaseMixin, unittest.TestCase):
face_normals_cpu[nonzero], face_normals_cuda.cpu()[nonzero], atol=1e-6
)
def test_equality(self):
meshes1 = init_mesh(num_meshes=2)
meshes2 = init_mesh(num_meshes=2)
meshes3 = init_mesh(num_meshes=3)
empty_mesh = Meshes([], [])
self.assertTrue(mesh_structures_equal(empty_mesh, Meshes([], [])))
self.assertTrue(mesh_structures_equal(meshes1, meshes1))
self.assertTrue(mesh_structures_equal(meshes1, meshes1.clone()))
self.assertFalse(mesh_structures_equal(empty_mesh, meshes1))
self.assertFalse(mesh_structures_equal(meshes1, meshes2))
self.assertFalse(mesh_structures_equal(meshes1, meshes3))
@staticmethod
def compute_packed_with_init(
num_meshes: int = 10, max_v: int = 100, max_f: int = 300, device: str = "cpu"