PoseNet
Pytorch PoseNet implementation based on paper PoseNet: A Convolutional Network for Real-Time 6-DOF Camera Relocalization
Dependencies
- PyTorch, NumPy, tqdm for PoseNet
- Pandas for custom dataset used
Files
- experiments_unclean contains unstructured code used during experiments
- main contains structured code, PoseNet class, and usage example
Notes
Both experiments and model were created & tested in Google Colab, so it's likely that some changes needed to run this code locally (at least in data paths)