目录
2021SC@SDUSC
2021SC@SDUSC
中心损失函数及实现代码
中心损失主要主要用于减少类内距离,虽然只是减少了累内距离,效果上却可以表现出累内距离小了,类间距离就可以增大的效果。该损失不可以直接使用,需要配合传统的softmax loss一起使用。可以起到比单纯softmax loss更好的分类效果。
实现代码:
在训练过程中,保存当前所有类中心的全连接预测特征,即代码中centers,每个batch的计算需要先读取已经保存的centers。
def center_loss(features, label, alfa, nrof_classes):
"""Center loss based on the paper "A Discriminative Feature Learning Approach for Deep Face Recognition"
(http://ydwen.github.io/papers/WenECCV16.pdf)
https://blog.csdn.net/u014380165/article/details/7694633