pose encoder trial

2023-06-07 19:03:53 +00:00
parent 9f7444f1e4
commit 5809f5a68a
3 changed files with 16 additions and 16 deletions
@@ -30,8 +30,8 @@ from pytorch3d.structures import Meshes
 class PoseRenderer(Module):
    def __init__(self, device='cuda') -> None:
        super().__init__()
-        # colors = np.load('custom_data/pose_colors.npy')
-        colors = np.load('custom_data/body_colors_interpolated_unweighted.npy')
+        colors = np.load('custom_data/pose_colors.npy')
+        # colors = np.load('custom_data/body_colors_interpolated.npy').astype(np.float32) / 255.0
        
        self.textures = TexturesVertex(verts_features=torch.from_numpy(colors).to(device)[None, :, :])
        self.cam = torch.tensor([3.0, 0.0, 0.0], device=device).cpu().numpy()