sold2 / README.md
ducha-aiki's picture
Update README
27ff5cf verified
---
license: mit
tags:
- kornia
- line-detection
- feature-description
---
# kornia/sold2
Pretrained weights for **SOLD²** (Self-supervised Occlusion-aware Line Description
and Detection), used by
[`kornia.feature.SOLD2`](https://kornia.readthedocs.io/en/latest/feature.html).
SOLD² detects line segments and computes semi-dense descriptors along them using a
shared encoder. Trained on wireframe and outdoor datasets. CVPR 2021.
**Original repo:** [cvg/SOLD2](https://github.com/cvg/SOLD2)
## Weights
| File | Training data |
|------|--------------|
| `sold2_wireframe.pth` | Wireframe dataset |
## Citation
```bibtex
@inproceedings{SOLD22021,
author = {Pautrat*, Rémi and Lin*, Juan-Ting and Larsson, Viktor
and Oswald, Martin R. and Pollefeys, Marc},
title = {{SOLD2}: Self-supervised Occlusion-aware Line Description and Detection},
booktitle = {CVPR},
year = {2021}
}
```