Fix Typo

Fix Typo : h36m joint shape (24->17)
microsoft · Aug 31, 2021 · 0bb35b6 · 0bb35b6
1 parent 30ea651
commit 0bb35b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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