pytorch-metric-learning
pytorch-metric-learning copied to clipboard
Selectively Contrastive Triplet loss
Hi Kevin
Would you mind add my Selectively Contrastive Triplet loss (which is published in ECCV2020)? The major idea of this paper is to overcome the local minima during the triplet optimization. This loss work especially well on high intra-variance datasets such as Hotel50K. And I provide my implementation in this repo:
https://github.com/littleredxh/HardNegative
You can find the loss function in "_code/Loss.py" file
I think this loss function can be easily adapted into your framework.
👍 Will add to my to-do list
Can I commit to this feature? I mean to be a contributor? Or It's almost ready?
Please do. I never got around to working on it.