save all

2023-08-24 11:36:22 +00:00
parent 59f67a2495
commit afd482522d
36 changed files with 2194 additions and 291 deletions
@@ -35,6 +35,8 @@ def fix_collision(V: torch.Tensor, B: torch.Tensor, mask: torch.Tensor, thr=.004
        F_fix_coll_local = F_fix_coll_orig
    else:
        raise ValueError("V must have 14475 or 3638 vertices")
+    if V.shape[0] == 1:
+        F_fix_coll_local = F_fix_coll_local[:1]
    mesh = Meshes(B, F_fix_coll_local)
    B_vn = torch.stack(mesh.verts_normals_list())