KaiyangZhou/pytorch-center-loss
PyTorch implementation of Center Loss for deep face recognition, based on the ECCV 2016 paper by Wen et al.

Velocity · 7d
+0.3
★ / day
Trend
→steady
star history
This repository provides a PyTorch implementation of Center Loss, a loss function designed to enhance discriminative feature learning in deep neural networks for face recognition. The approach learns a center for each class and penalizes the distance between deep features and their class centers. It is demonstrated here on MNIST classification but is also used in larger projects like deep-person-reid for person re-identification.