Chenwei-Huang / PIR-Net

Demo code of the paper "PIR-Net: Pixel-wise Image Registration Network for RGB Images"

PIR-Net

Demo code of the paper "PIR-Net: Pixel-wise Image Registration Network for RGB Images"

Requirements

torch>=0.3
torchvision
imageio
argparse
scipy
opencv-python 4.6.2

How to run

• 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"

Visualization

We also provide a visualization demo (transform.m) to test our registration model.