TinyZeaMays / CircleLoss

Pytorch implementation of the paper "Circle Loss: A Unified Perspective of Pair Similarity Optimization"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to obtain class labels?

AddASecond opened this issue · comments

I wonder how to obtain class labels in classification tasks?
I have checked circle_loss.py but it only learn similarity between MNIST images

Yes, the circle loss is designed for metric learning. CircleLossLikeCE (in circle_loss_early.py, which is my implementation only and does not consist with the paradim of circle loss) is designed for classificasion in the paradim of approaches like ArcFace, but I'm not sure any improvements could be acheived.