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

Fix AttributeError

Browse Source 
Fix AttributeError: 'SMPLlayer' object has no attribute 'NUM_EXPR'
This commit is contained in:
532stary4 2021-10-17 17:10:40 +01:00
parent 2544c970ef
commit 73c7a00fad
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)), 'Th': np.zeros((nFrames, 3)),
} }
if self.model_type == 'smplx': if self.model_type == 'smplx':
params['expression'] = np.zeros((nFrames, self.NUM_EXPR)) params['expression'] = np.zeros((nFrames, NUM_EXPR))
if ret_tensor: if ret_tensor:
for key in params.keys(): for key in params.keys():
params[key] = to_tensor(params[key], self.dtype, self.device) 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: 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])))) 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(): 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 return body_params
@staticmethod @staticmethod