Demo code of the paper "PIR-Net: Pixel-wise Image Registration Network for RGB Images"
torch>=0.3
torchvision
imageio
argparse
scipy
opencv-python 4.6.2
- Download MVR dataset (code: vbnm) and Extract the data under the project.
- Download pretrained model (code: vbnm) and Extract the file under the project.
- run "python test.py"
We also provide a visualization demo (transform.m) to test our registration model.