(self, joints, **kwargs)
| 126 | torch.tensor(joint_maps, dtype=torch.long)) |
| 127 | |
| 128 | def forward(self, joints, **kwargs): |
| 129 | if self.joint_maps is None: |
| 130 | return joints |
| 131 | else: |
| 132 | return torch.index_select(joints, 1, self.joint_maps) |
| 133 | # import smplx |
| 134 | # smplx_model_path = r'C:\Users\lithiumice\code\smplify-x\models' |
| 135 | # joint_mapper = JointMapper( |
nothing calls this directly
no outgoing calls
no test coverage detected