Spaces:
Running
Running
File size: 2,066 Bytes
a80d6bb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
# 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](network_v0) and [network_v1](network_v1), the corresponding pre-trained models are in [checkpoints](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](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}
}
```
## Related Projects
https://github.com/TRI-ML/KP2D
|