(self, joint_maps=None)
| 118 | |
| 119 | class JointMapper(nn.Module): |
| 120 | def __init__(self, joint_maps=None): |
| 121 | super(JointMapper, self).__init__() |
| 122 | if joint_maps is None: |
| 123 | self.joint_maps = joint_maps |
| 124 | else: |
| 125 | self.register_buffer('joint_maps', |
| 126 | torch.tensor(joint_maps, dtype=torch.long)) |
| 127 | |
| 128 | def forward(self, joints, **kwargs): |
| 129 | if self.joint_maps is None: |