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}
}