Rethinking Low-level Features for Interest Point Detection and Description
Dependency
- pytorch
- torchvision
- cv2
- tqdm
We use cuda 11.4/python 3.8.13/torch 1.10.0/torchvision 0.11.0/opencv 3.4.8 for training and testing.
Pre-trained models
We provide two versions of LANet with different structure in network_v0 and network_v1, the corresponding pre-trained models are in checkpoints.
- v0: The original version used in our paper.
- v1: An improved version that has a better over all performance.
Training
Download the COCO dataset:
cd datasets/COCO/
wget http://images.cocodataset.org/zips/train2017.zip
unzip train2017.zip
Prepare the training file:
python datasets/prepare_coco.py --raw_dir datasets/COCO/train2017/ --saved_dir datasets/COCO/
To train the model (v0) on COCO dataset, run:
python main.py --train_root datasets/COCO/train2017/ --train_txt datasets/COCO/train2017.txt
Evaluation
Evaluation on HPatches dataset
Download the HPatches dataset:
cd datasets/HPatches/
wget http://icvl.ee.ic.ac.uk/vbalnt/hpatches/hpatches-sequences-release.tar.gz
tar -xvf hpatches-sequences-release.tar.gz
To evaluate the pre-trained model, run:
python test.py --test_dir ./datasets/HPatches/hpatches-sequences-release
License
The code is released under the MIT license.
Citation
Please use the following citation when referencing our work:
@InProceedings{Wang_2022_ACCV,
author = {Changhao Wang and Guanwen Zhang and Zhengyun Cheng and Wei Zhou},
title = {Rethinking Low-level Features for Interest Point Detection and Description},
booktitle = {Computer Vision - {ACCV} 2022 - 16th Asian Conference on Computer
Vision, Macao, China, December 4-8, 2022, Proceedings, Part {II}},
series = {Lecture Notes in Computer Science},
volume = {13842},
pages = {108--123},
year = {2022}
}