Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

adding more features to SMPLX model #50

Open
orydatadudes opened this issue Oct 21, 2021 · 0 comments
Open

adding more features to SMPLX model #50

orydatadudes opened this issue Oct 21, 2021 · 0 comments

Comments

@orydatadudes
Copy link

when init the class SMPLXLayer and adding more features - changing
False to True

class SMPLXLayer(SMPLX):
def init(
self,
*args,
**kwargs
) -> None:

    super(SMPLXLayer, self).__init__(
        create_global_orient=False,
        create_body_pose=False,
        create_left_hand_pose=False,
        create_right_hand_pose=False,
        create_jaw_pose=False,
        create_leye_pose=False,
        create_reye_pose=False,
        create_betas=False,
        create_expression=False,
        create_transl=False,
        *args, **kwargs,
    )

i am getting that error

Traceback (most recent call last):
File "expose-master/demo.py", line 578, in
rcnn_batch=rcnn_batch,
File "/home/anaconda3/envs/temp/lib/python3.7/site-packages/torch/autograd/grad_mode.py", line 28, in decorate_context
return func(*args, **kwargs)
File "/expose-master/demo.py", line 245, in main
model = SMPLXNet(exp_cfg)
File "expose-master/expose/models/smplx_net.py", line 39, in init
self.smplx = build_attention_head(exp_cfg)
File "expose-master/expose/models/attention/build.py", line 23, in build_attention_head
return SMPLXHead(cfg)
File "expose-master/expose/models/attention/predictor.py", line 251, in init
**body_model_cfg)
File "/home/anaconda3/envs/temp/lib/python3.7/site-packages/smplx/body_models.py", line 2342, in build_layer
return SMPLXLayer(model_path, **kwargs)
File "/home/anaconda3/envs/temp/lib/python3.7/site-packages/smplx/body_models.py", line 1310, in init
*args, **kwargs,
File "/home/anaconda3/envs/temp/lib/python3.7/site-packages/smplx/body_models.py", line 991, in init
**kwargs)
File "/home/anaconda3/envs/temp/lib/python3.7/site-packages/smplx/body_models.py", line 596, in init
use_compressed=use_compressed, dtype=dtype, ext=ext, **kwargs)
File "/home/anaconda3/envs/temp/lib/python3.7/site-packages/smplx/body_models.py", line 206, in init
global_orient, dtype=dtype)
TypeError: must be real number, not CfgNode

Process finished with exit code 1

the error happen during execute :

default_global_orient = torch.tensor(
global_orient, dtype=dtype)

global_orient value is : CfgNode({'param_type': 'cont_rot_repr'})

the same is relevant for the rest of the features (create_body_pose,create_left_hand_pose..)
thank you

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant