Skip to content

sailor-z/NM-Net

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

34 Commits
 
 
 
 
 
 
 
 

Repository files navigation

NM-Net: Mining Reliable Neighbors for Robust Feature Correspondences (CVPR 2019 oral)


This repository is a reference implementation for "NM-Net: Mining Reliable Neighbors for Robust Feature Correspondences", CVPR 2019 oral. If you use this code in your research, please cite the paper.

@inproceedings{zhao2019nm,
  title={NM-Net: Mining reliable neighbors for robust feature correspondences},
  author={Zhao, Chen and Cao, Zhiguo and Li, Chi and Li, Xin and Yang, Jiaqi},
  booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
  pages={215--224},
  year={2019}
}

Installation


pip install -r requirements.txt

Preparing data


Edit config.data_tr in config.py to prepare data for different datasets.

python ./dump_data.py

Training


For the first time running main.py in each dataset, set the parameter initialize in Data_Loader to be True.

python ./main.py COLMAP #NARROW WIDE

End-to-end version


An end-to-end version has been released which is independent to affine information. Please refer to code/NM_Net_v2.py.

Acknowledgement


The data processing and evaluation codes are borrowed from "Learning to Find Good Correspondences" (CVPR 2018). Please cite this paper if the code is useful for your research.

@inproceedings{moo2018learning,
  title={Learning to find good correspondences},
  author={Moo Yi, Kwang and Trulls, Eduard and Ono, Yuki and Lepetit, Vincent and Salzmann, Mathieu and Fua, Pascal},
  booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
  pages={2666--2674},
  year={2018}
}

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published