sosnet / README.md
ducha-aiki's picture
Update README
73f0282 verified
metadata
license: mit
tags:
  - kornia
  - feature-description
  - patch-descriptor

kornia/sosnet

Pretrained weights for SOSNet, used by kornia.feature.SOSNet.

SOSNet is a 128-dimensional patch descriptor that adds a second-order similarity regularisation term to the HardNet loss, enforcing that the structure of the descriptor space reflects patch similarity at multiple scales. CVPR 2019.

Original repo: yuruntian/SOSNet

Weights

File Training data
sosnet_32x32_liberty.pth Liberty
sosnet_32x32_hpatches_a.pth HPatches-A

Citation

@InProceedings{sosnet2019,
    author    = {Tian, Yurong and Yu, Xin and Fan, Bin and Wu, Fuchao
                 and Heijnen, Huub and Balntas, Vassileios},
    title     = {{SOSNet}: Second Order Similarity Regularization for
                 Local Descriptor Learning},
    booktitle = {CVPR},
    year      = {2019}
}