Chenwei-Huang/DPH-Net

Demo code of the paper "Deep Image Registration With Depth-Aware Homography Estimation"

Python

DPH-Net

Demo code of the paper "Deep Image Registration With Depth-Aware Homography Estimation"

Requirements

pytorch 1.6.0
cuda 10.2
opencv-python 4.5.1

Preparation

Download IRD dataset (code: asdf) and Extract the data under the project.
Download pretrained model (code: asdf) and Extract the file under the project.

Test

run "sh command_test.sh"

Train

run "sh command_train.sh"