pyramid pipeline ready

2023-05-22 00:13:39 +02:00
parent 8e0d061104
commit 52e461ca47
6 changed files with 938 additions and 793 deletions
@@ -25,8 +25,7 @@ to_keep = sio.loadmat('smpl/keep_remove.mat')['keep'].reshape(-1)
 F_fix_coll = torch.tensor(np.tile(np.load("smpl/smpl_faces.npy")[to_keep,:][None,:,:], [4,1,1]), device='cuda')


-
-def fix_collision(V, B, mask, thr=.004):
+def fix_collision(V: torch.Tensor, B: torch.Tensor, mask: torch.Tensor, thr=.004):
    mesh = Meshes(B, F_fix_coll)
    B_vn = torch.stack(mesh.verts_normals_list())