fulyvv/EasyMocap
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🐛 fix basic

Browse Source
This commit is contained in:
shuaiqing 2021-07-02 11:48:41 +08:00
parent b2d5bbcfd7
commit 72d96db48b
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
easymocap/pipeline/basic.py
View File

@ -2,7 +2,7 @@
@ Date: 2021-04-13 20:43:16 @ Date: 2021-04-13 20:43:16
@ Author: Qing Shuai @ Author: Qing Shuai
@ LastEditors: Qing Shuai @ LastEditors: Qing Shuai
@ LastEditTime: 2021-05-27 15:35:22 @ LastEditTime: 2021-06-14 15:32:24
@ FilePath: /EasyMocap/easymocap/pipeline/basic.py @ FilePath: /EasyMocap/easymocap/pipeline/basic.py
''' '''
from ..pyfitting import optimizeShape, optimizePose2D, optimizePose3D from ..pyfitting import optimizeShape, optimizePose2D, optimizePose3D
@ -57,7 +57,7 @@ def smpl_from_keypoints3d2d(body_model, kp3ds, kp2ds, bboxes, Pall, config, args
model_type = body_model.model_type model_type = body_model.model_type
params_init = body_model.init_params(nFrames=1) params_init = body_model.init_params(nFrames=1)
if weight_shape is None: if weight_shape is None:
weight_shape = load_weight_shape(args.opts) weight_shape = load_weight_shape(model_type, args.opts)
if model_type in ['smpl', 'smplh', 'smplx']: if model_type in ['smpl', 'smplh', 'smplx']:
# when use SMPL model, optimize the shape only with first 1-14 limbs, # when use SMPL model, optimize the shape only with first 1-14 limbs,
# don't use (nose, neck) # don't use (nose, neck)
@ -82,7 +82,7 @@ def smpl_from_keypoints3d(body_model, kp3ds, config, args,
model_type = body_model.model_type model_type = body_model.model_type
params_init = body_model.init_params(nFrames=1) params_init = body_model.init_params(nFrames=1)
if weight_shape is None: if weight_shape is None:
weight_shape = load_weight_shape(args.opts) weight_shape = load_weight_shape(model_type, args.opts)
if model_type in ['smpl', 'smplh', 'smplx']: if model_type in ['smpl', 'smplh', 'smplx']:
# when use SMPL model, optimize the shape only with first 1-14 limbs, # when use SMPL model, optimize the shape only with first 1-14 limbs,
# don't use (nose, neck) # don't use (nose, neck)