diff --git a/phalp/models/hmar/hmar.py b/phalp/models/hmar/hmar.py index 7044db1..1a584d0 100755 --- a/phalp/models/hmar/hmar.py +++ b/phalp/models/hmar/hmar.py @@ -36,7 +36,7 @@ def __init__(self, cfg): self.texture_head = TextureHead(self.uv_sampler, self.cfg, img_H=img_H, img_W=img_W) self.encoding_head = EncodingHead(cfg=self.cfg, img_H=img_H, img_W=img_W) - smpl_cfg = {k.lower(): v for k,v in dict(cfg.SMPL).items()} + smpl_cfg = {k.lower(): v for k,v in cfg.SMPL.__dict__.items()} self.smpl = SMPL(**smpl_cfg) self.smpl_head = SMPLHead(cfg, @@ -130,4 +130,4 @@ def get_3d_parameters(self, pred_smpl_params, pred_cam, center=np.array([128, 12 pred_keypoints_2d_smpl = (pred_keypoints_2d_smpl+0.5)*img_size - return pred_smpl_params, pred_keypoints_2d_smpl, pred_joints, pred_cam_t \ No newline at end of file + return pred_smpl_params, pred_keypoints_2d_smpl, pred_joints, pred_cam_t