[ECCV2024] Diff-Reg: Diffusion Model in Doubly Stochastic Matrix Space for Registration Problem [Arxiv|ECCV(Todo)]
👀 If you have any questions, please let me (wuqianliang@njust.edu.cn) know~
1. Please use the NVIDIA TITAN RTX or NVIDIA GeForce RTX 3090 GPU !!
2. Please utilize commands 'conda env create -f Diff-Reg-2d3d/eccv24_2d3d_env.yml', 'conda env create -f Diff-Reg-3dmatch/eccv24_3d_env.yml', and 'conda env create -f Diff-Reg-4dmatch/eccv24_4d_env.yml' to install environments for three tasks.
Please look at the release page for the pre-trained model weights of three experiments.
Our 2D3D registration code is mainly based on 2D3D-MATR, and our 3d registration code is based on Lepard. Please refer to Lepard and 2D3D-MATR.
We thank the respective authors of Lepard,2D3D-MATR, GeoTR,RoITR,GraphSCNet, and Vision3D for their open source code.
Please consider citing the following BibTeX entry if you find our work helpful for your research.
@article{wu2024diff,
title={Diff-Reg v1: Diffusion Matching Model for Registration Problem},
author={Wu, Qianliang and Jiang, Haobo and Luo, Lei and Li, Jun and Ding, Yaqing and Xie, Jin and Yang, Jian},
journal={arXiv preprint arXiv:2403.19919},
year={2024}
}