onlyNata / Deep-homography-estimation-pytorch

PyTorch implementation of Deep Homography Estimation.

Deep Image Homography Estimation- PyTorch Implementation

Deep Image Homography Estimation
Daniel DeTone, Tomasz Malisiewicz, and Andrew Rabinovich

Generate training dataset

cd data/
python gen_data.py

Training

python train.py

Test

Download pre-trained weights

链接：https://pan.baidu.com/s/10HXNthOBhlZbrtvIkolxKw 	提取码：l9l8 

Store the model to checkpoints/ folder

python test.py

results

Reference

https://github.com/mazenmel/Deep-homography-estimation-Pytorch