Alexadlu / NM-Net

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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


This repository is a reference implementation for Chen Zhao, Zhiguo Cao, Chi Li, Xin Li, and Jiaqi Yang, "NM-Net: Mining Reliable Neighbors for Robust Feature Correspondences", CVPR 2019 oral. If you use this code in your research, please cite the paper.

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

About


Languages

Language:C 59.3%Language:MATLAB 17.7%Language:Python 9.0%Language:HTML 6.6%Language:Makefile 3.0%Language:CSS 1.4%Language:Objective-C 1.2%Language:C++ 0.7%Language:Roff 0.5%Language:TeX 0.3%Language:Clean 0.3%Language:Shell 0.1%Language:M 0.1%