kgoodridge / keras-posenet

Implementation of the PoseNet Architecture in Keras

Keras-PoseNet

This is an implementation for Keras of 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/]

Note that this requires the TensorFlow backend for Keras, however, only minor modifications to the model as well as the helper.py file would be required in order to use the Theano backend. If someone would like assistance with this, simply open an issue.

Getting Started

• Download the Cambridge Landmarks King's College dataset from here.

• Download the starting and trained weights from here.

• The PoseNet model is defined in the posenet.py file

• The starting and trained weights (posenet.npy and trained_weights.h5 respectively) for training were obtained by converting caffemodel weights from here and then training.

• To run:

  • Extract the King's College dataset to wherever you prefer
  • Extract the starting and trained weights to the same location as test.py and train.py
  • Update the dataset path on line 9 in helper.py
  • If you want to retrain, simply run train.py (note this will take a long time)
  • If you just want to test, simply run test.py