search

KevinMusgrave / pytorch-metric-learning

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
https://kevinmusgrave.github.io/pytorch-metric-learning/
MIT License
6.03k stars 658 forks source link

center loss implementation #108

Open amirhfarzaneh opened 4 years ago

amirhfarzaneh commented 4 years ago

Is center loss considered to be a metric learning approach? If you think it can be part of this repo, I am interested in contributing.

KevinMusgrave commented 4 years ago

Yes that would be great! I assume you're talking about this paper. In terms of implementation, take a look at NormalizedSoftmaxLoss for an example of how a classification loss is implemented.

amirhfarzaneh commented 4 years ago

Yes it's the same paper. I will take a look at that example.