File size: 1,013 Bytes
2f4a08d | 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 | ---
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`](https://kornia.readthedocs.io/en/latest/feature.html).
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](https://github.com/cvlab-epfl/disk)
## Weights
| File | Training supervision |
|------|----------------------|
| `depth-save.pth` | Depth |
| `epipolar-save.pth` | Epipolar geometry |
## Citation
```bibtex
@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}
}
```
|