pose_feat: BxC preliminary_cameras: cameras in opencv coordinate.
(
self,
pose_feat,
)
| 31 | ) |
| 32 | |
| 33 | def forward( |
| 34 | self, |
| 35 | pose_feat, |
| 36 | ): |
| 37 | """ |
| 38 | pose_feat: BxC |
| 39 | preliminary_cameras: cameras in opencv coordinate. |
| 40 | """ |
| 41 | |
| 42 | pred_cameras = self.mlp(pose_feat) # Bx7, 3 for absT, 4 for quaR |
| 43 | return pred_cameras |
| 44 | |
| 45 | |
| 46 | class PoseEncoder(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected