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

lint fixes

Browse Source 
Summary:
Ran the linter.
TODO: need to update the linter as per D21353065.

Reviewed By: bottler

Differential Revision: D21362270

fbshipit-source-id: ad0e781de0a29f565ad25c43bc94a19b1828c020
This commit is contained in:
Nikhila Ravi
2020-05-04 09:55:03 -07:00
committed by Facebook GitHub Bot
parent 0c595dcf5b
commit 0eca74fa5f
15 changed files with 73 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_mesh_normal_consistency.py
View File

@@ -180,7 +180,7 @@ class TestMeshNormalConsistency(unittest.TestCase):
# mesh1: normal consistency computation
n0 = (verts1[1] - verts1[2]).cross(verts1[3] - verts1[2])
n1 = (verts1[1] - verts1[2]).cross(verts1[0] - verts1[2])
loss1 = 1.0 - torch.cosine_similarity(n0.view(1, 3), -n1.view(1, 3))
loss1 = 1.0 - torch.cosine_similarity(n0.view(1, 3), -(n1.view(1, 3)))
# mesh2: normal consistency computation
# In the cube mesh, 6 edges are shared with coplanar faces (loss=0),
@@ -193,9 +193,9 @@ class TestMeshNormalConsistency(unittest.TestCase):
n2 = (verts3[1] - verts3[2]).cross(verts3[4] - verts3[2])
loss3 = (
3.0
- torch.cosine_similarity(n0.view(1, 3), -n1.view(1, 3))
- torch.cosine_similarity(n0.view(1, 3), -n2.view(1, 3))
- torch.cosine_similarity(n1.view(1, 3), -n2.view(1, 3))
- torch.cosine_similarity(n0.view(1, 3), -(n1.view(1, 3)))
- torch.cosine_similarity(n0.view(1, 3), -(n2.view(1, 3)))
- torch.cosine_similarity(n1.view(1, 3), -(n2.view(1, 3)))
)
loss3 /= 3.0