Sailor-z/NM-Net - GitHub

Trang chủ » Nm Github » Sailor-z/NM-Net - GitHub

Có thể bạn quan tâm

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} }

Từ khóa » Nm Github