/PIR-Net

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

Primary LanguagePython

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.