metadata
license: mit
tags:
- kornia
- feature-detection
- feature-description
- image-matching
kornia/disk
Pretrained weights for DISK (Learning local features with policy gradient),
used by kornia.feature.DISK.
DISK is trained with policy gradient and produces dense heatmaps of keypoint probability and per-pixel 128-dim descriptors via a U-Net backbone. NeurIPS 2020.
Original repo: cvlab-epfl/disk
Weights
| File | Training supervision |
|---|---|
depth-save.pth |
Depth |
epipolar-save.pth |
Epipolar geometry |
Citation
@article{tyszkiewicz2020disk,
title = {{DISK}: Learning local features with policy gradient},
author = {Tyszkiewicz, Micha{\l} and Fua, Pascal and Trulls, Eduard},
journal = {Advances in Neural Information Processing Systems},
volume = {33},
pages = {14254--14265},
year = {2020}
}