| 631 | self.register_buffer('pose_mean', pose_mean_tensor) |
| 632 | |
| 633 | def create_mean_pose(self, data_struct, flat_hand_mean=False): |
| 634 | # Create the array for the mean pose. If flat_hand is false, then use |
| 635 | # the mean that is given by the data, rather than the flat open hand |
| 636 | global_orient_mean = torch.zeros([3], dtype=self.dtype) |
| 637 | body_pose_mean = torch.zeros([self.NUM_BODY_JOINTS * 3], |
| 638 | dtype=self.dtype) |
| 639 | |
| 640 | pose_mean = torch.cat([ |
| 641 | global_orient_mean, body_pose_mean, self.left_hand_mean, |
| 642 | self.right_hand_mean |
| 643 | ], |
| 644 | dim=0) |
| 645 | return pose_mean |
| 646 | |
| 647 | def name(self) -> str: |
| 648 | return 'SMPL+H' |