vhad/torch-mesh-isect
1
0
Fork 0
mirror of https://github.com/gosticks/torch-mesh-isect.git synced 2025-10-16 11:55:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix bug in gradient computation

Browse Source
This commit is contained in:
Mohamed Hassan 2019-12-10 18:21:36 +01:00
parent 9b6ab62310
commit 3b3552e373
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
mesh_intersection/loss.py
View File

@ -254,14 +254,14 @@ class DistanceFieldPenetrationLoss(nn.Module):
recv_triangles = triangles[batch_idxs, receiver_faces] recv_triangles = triangles[batch_idxs, receiver_faces]
intr_triangles = triangles[batch_idxs, intruder_faces] intr_triangles = triangles[batch_idxs, intruder_faces]
with torch.no_grad():
recv_normals = normals[batch_idxs, receiver_faces] recv_normals = normals[batch_idxs, receiver_faces]
recv_circumradius = circumradius[batch_idxs, receiver_faces] recv_circumradius = circumradius[batch_idxs, receiver_faces]
recv_circumcenter = circumcenter[batch_idxs, receiver_faces] recv_circumcenter = circumcenter[batch_idxs, receiver_faces]
intr_normals = normals[batch_idxs, intruder_faces] intr_normals = normals[batch_idxs, intruder_faces]
intr_circumradius = circumradius[batch_idxs, intruder_faces] intr_circumradius = circumradius[batch_idxs, intruder_faces]
intr_circumcenter = circumcenter[batch_idxs, intruder_faces] intr_circumcenter = circumcenter[batch_idxs, intruder_faces]
# Compute the distance field for the intruding triangles # Compute the distance field for the intruding triangles
# B x NUM_COLLISIONS x 3 # B x NUM_COLLISIONS x 3