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

Merge pull request #93 from 532stary4/pull-branch

Browse Source 
Fix AttributeError
This commit is contained in:
Qing Shuai 2021-10-26 10:35:38 +08:00 committed by GitHub
commit e7bcf6b3ed
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
easymocap/smplmodel/body_model.py
View File

@ -378,7 +378,7 @@ class SMPLlayer(nn.Module):
'Th': np.zeros((nFrames, 3)),
}
if self.model_type == 'smplx':
params['expression'] = np.zeros((nFrames, self.NUM_EXPR))
params['expression'] = np.zeros((nFrames, NUM_EXPR))
if ret_tensor:
for key in params.keys():
params[key] = to_tensor(params[key], self.dtype, self.device)
@ -390,7 +390,7 @@ class SMPLlayer(nn.Module):
if body_params['poses'].shape[1] != self.NUM_POSES:
body_params['poses'] = np.hstack((body_params['poses'], np.zeros((nFrames, self.NUM_POSES - body_params['poses'].shape[1]))))
if model_type == 'smplx' and 'expression' not in body_params.keys():
body_params['expression'] = np.zeros((nFrames, self.NUM_EXPR))
body_params['expression'] = np.zeros((nFrames, NUM_EXPR))
return body_params
@staticmethod