← all repositories

KaiyangZhou/pytorch-center-loss

PyTorch implementation of Center Loss for deep face recognition, based on the ECCV 2016 paper by Wen et al.

994 stars Python Computer VisionML Frameworks
pytorch-center-loss
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.