fix

open-mmlab · Nov 2, 2023 · 273f2a6 · 273f2a6
1 parent 47010c7
commit 273f2a6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mmhuman3d/data/data_converters/agora.py b/mmhuman3d/data/data_converters/agora.py
@@ -212,7 +212,8 @@ def convert_by_mode(self, dataset_path: str, out_path: str,
 
             if smplx_param['betas'].shape[1] != 10:
                 smplx_param['betas_extra'] = smplx_param['betas'][:, 10:]
-                smplx_param['betas'] = smplx_param['betas_neutral'][:, :10]
+                smplx_param['betas'] = smplx_param['betas'][:, :10]
+                smplx_param['betas_neutral'] = smplx_param['betas_neutral'][:, :10]
             else:
                 smplx_param['betas_extra'] = np.zeros([1, 1])
             for key in self.smplx_shape.keys():