KaiyangZhou/pytorch-center-loss
Pytorch implementation of Center Loss
This project provides an implementation of the Center Loss function, a technique used to improve the accuracy of deep learning models in classification tasks. It takes raw data, such as images, and helps the model learn more distinct features for each class. The output is a more robust classification model, particularly useful for researchers and practitioners working on tasks like face recognition or person re-identification.
995 stars. No commits in the last 6 months.
Use this if you are a deep learning researcher or practitioner looking to enhance the discriminative power of your classification models, especially for tasks requiring fine-grained distinction between categories.
Not ideal if you are looking for a plug-and-play solution for general image classification without diving into custom loss function implementations.
Stars
995
Forks
220
Language
Python
License
MIT
Category
Last pushed
Feb 19, 2023
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/KaiyangZhou/pytorch-center-loss"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
HelmchenLabSoftware/Cascade
Calibrated inference of spiking from calcium ΔF/F data using deep networks
adobe/antialiased-cnns
pip install antialiased-cnns to improve stability and accuracy
kimhc6028/relational-networks
Pytorch implementation of "A simple neural network module for relational reasoning" (Relational Networks)
devsisters/pointer-network-tensorflow
TensorFlow implementation of "Pointer Networks"
kyegomez/LiqudNet
Implementation of Liquid Nets in Pytorch