Implementation of HomographyNet in Keras
- Contains
generate.py
, a script for generating a suitable dataset for the HomographyNet model. - A Jupyter notebook with general info about the structure of the generated dataset.
- Jupyter notebook with the training/test results
- Raw image files used
- Utility code used in the demo
- Training history in JSON format
A Python package containing the implementation of the HomographyNet model in Keras.
homographynet_weights_tf_dim_ordering_tf_kernels.h5
- pretrained weights of the baseline HomographyNet modelmobile_homographynet_weights_tf_dim_ordering_tf_kernels.h5
- pretrained weights of the MobileNet-based model
test.py
- a script for evaluating a specified model, or the pretrained model. It will download and cache the weights of the pretrained model on first use.
train.py
- a script for training the HomographyNet model