From 970bd5cd9d83d88995ea300d5e1f40bf3fc6fea3 Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Thu, 1 Jun 2023 23:09:09 +0200 Subject: [PATCH] collision and normal loss fix!!! --- losses/losses.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/losses/losses.py b/losses/losses.py index 4f47a09..5c59dfc 100644 --- a/losses/losses.py +++ b/losses/losses.py @@ -18,12 +18,12 @@ from torch_cluster import knn smpl_to_keep_orig = sio.loadmat('smpl/keep_remove.mat')['keep'].reshape(-1) smpl_to_keep_orig = smpl_to_keep_orig[smpl_to_keep_orig < 6890] -to_keep = sio.loadmat('smpl/keep_remove.mat')['keep'].reshape(-1) -F_coll = torch.tensor(np.load("smpl/smpl_faces.npy")[to_keep,:], device='cuda') -F_coll_skirt = torch.tensor(np.load("smpl/smpl_faces_skirt.npy")[to_keep,:], device='cuda') +# to_keep = sio.loadmat('smpl/keep_remove.mat')['keep'].reshape(-1) +F_coll = torch.tensor(np.load("smpl/smpl_faces.npy"), device='cuda') +F_coll_skirt = torch.tensor(np.load("smpl/smpl_faces_skirt.npy"), device='cuda') F_coll_orig = torch.tensor(np.load("smpl/smpl_faces_orig.npy")[:6890,:], device='cuda') -F_norm = torch.tensor(np.tile(np.load("smpl/smpl_faces.npy")[to_keep,:][None,:,:], [4,1,1]), device='cuda') -F_norm_skirt = torch.tensor(np.tile(np.load("smpl/smpl_faces_skirt.npy")[to_keep,:][None,:,:], [4,1,1]), device='cuda') +F_norm = torch.tensor(np.tile(np.load("smpl/smpl_faces.npy")[None,:,:], [4,1,1]), device='cuda') +F_norm_skirt = torch.tensor(np.tile(np.load("smpl/smpl_faces_skirt.npy")[None,:,:], [4,1,1]), device='cuda') F_norm_orig = torch.tensor(np.tile(np.load("smpl/smpl_faces_orig.npy")[:6890,:][None,:,:], [4,1,1]), device='cuda') F_norm_orig = F_norm_orig[:,smpl_to_keep_orig,:]