From 0bb35b601b32ec5a1464a41298547e370db8471f Mon Sep 17 00:00:00 2001 From: JooHo Kim Date: Wed, 1 Sep 2021 04:07:58 +0900 Subject: [PATCH] Fix Typo Fix Typo : h36m joint shape (24->17) --- metro/modeling/_smpl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metro/modeling/_smpl.py b/metro/modeling/_smpl.py index 92c90ca..bbfed17 100644 --- a/metro/modeling/_smpl.py +++ b/metro/modeling/_smpl.py @@ -132,7 +132,7 @@ def get_h36m_joints(self, vertices): Input: vertices: size = (B, 6890, 3) Output: - 3D joints: size = (B, 24, 3) + 3D joints: size = (B, 17, 3) """ joints = torch.einsum('bik,ji->bjk', [vertices, self.J_regressor_h36m_correct]) return joints