This is official Pytorch implementation of "C2RF: Bridging Multi-modal Image Registration and Fusion via Commonality Mining and Contrastive Learning". Please click here to download this paper.

• torch 1.10.2+cu102
• torchvision 0.8.2
• kornia 0.5.2

The framework of the proposed C2RF for multi-modal image registration and fusion.

Please download the pretrained weights at the link below, and then place them into the folder ./checkpoint/

• The pretrained weights for the Roadscene dataset is at Google Drive.

• The pretrained weights for the PET-MRI dataset is at Google Drive.

python test.py --dataset=RoadScene 

python test.py --dataset=PET-MRI

python train_Fu.py --dataset=RoadScene

python train_Fu.py --dataset=PET-MRI

python train_Reg.py --dataset=RoadScene

python train_Reg.py --dataset=PET-MRI