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