| 10 | |
| 11 | class PoseProjector(nn.Module): |
| 12 | def __init__(self, hidden_dim=256, num_body_points=17): |
| 13 | super().__init__() |
| 14 | self.num_body_points = num_body_points |
| 15 | self.V_projector = nn.Linear(hidden_dim, num_body_points) |
| 16 | nn.init.constant_(self.V_projector.bias.data, 0) |
| 17 | self.Z_projector = MLP(hidden_dim, hidden_dim, num_body_points * 2, 3) |
| 18 | nn.init.constant_(self.Z_projector.layers[-1].weight.data, 0) |
| 19 | nn.init.constant_(self.Z_projector.layers[-1].bias.data, 0) |
| 20 | |
| 21 | def forward(self, hs): |
| 22 | """_summary_ |