This is a deep convolutional neural network for estimating the relative homography between a pair of images. Deep Image Homography Estimation paper implementation in PyTorch.
- Backbone: MobileNetV2
- Dataset: MSCOCO 2014 training set
- Train/valid: generated 500,000/41,435 pairs of image patches sized 128x128(rho=32).
- Test: generated 10,000 pairs of image patches sized 256x256(rho=64).
- Python 3.6.8
- PyTorch 1.3.0
Extract training images:
$ python3 extract.py
$ python3 pre_process.py
$ python3 train.py --lr 0.005 --batch-size 64
If you want to visualize during training, run in your terminal:
$ tensorboard --logdir runs
Homography Estimation Comparison on Warped MS-COCO 14 Test Set.
$ python3 test.py
$ python3 test_orb.py --type surf
$ python3 test_orb.py --type identity
Method | Mean Average Corner Error (pixels) |
---|---|
HomographyNet | 3.53 |
SURF + RANSAC | 8.83 |
Identity Homography | 32.13 |