caffe-posenet

by alexgkendall

alexgkendall / caffe-posenet

Implementation of PoseNet

433 Stars 190 Forks Last release: Not found Other 3.5K Commits 0 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

PoseNet

This is a modified version of Caffe which supports the PoseNet architecture

As described in the ICCV 2015 paper PoseNet: A Convolutional Network for Real-Time 6-DOF Camera Relocalization Alex Kendall, Matthew Grimes and Roberto Cipolla [http://mi.eng.cam.ac.uk/projects/relocalisation/]

Getting Started

  • Download the Cambridge Landmarks dataset from here.
  • Download models and trained weights from here.

Create an LMDB localisation dataset with

caffe-posenet/posenet/scripts/create_posenet_lmdb_dataset.py
Change lines 1, 11 & 12 to the appropriate directories.

Test PoseNet with

caffe-posenet/posenet/scripts/test_posenet.py
using the command
python test_posenet.py --model your_model.prototxt --weights your_weights.caffemodel --iter size_of_dataset

Publications

If you use this software in your research, please cite our publications:

http://arxiv.org/abs/1505.07427 Alex Kendall, Matthew Grimes and Roberto Cipolla "PoseNet: A Convolutional Network for Real-Time 6-DOF Camera Relocalization." Proceedings of the International Conference on Computer Vision (ICCV), 2015.

http://arxiv.org/abs/1509.05909 Alex Kendall and Roberto Cipolla "Modelling Uncertainty in Deep Learning for Camera Relocalization." The International Conference on Robotics and Automation, 2015.

License

This extension to the Caffe library and the PoseNet models are released under a creative commons license which allows for personal and research use only. For a commercial license please contact the authors. You can view a license summary here: http://creativecommons.org/licenses/by-nc/4.0/

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.