vandit15/Class-balanced-loss-pytorch
vandit15/Class-balanced-loss-pytorch has 801 GitHub stars and 122 forks. GitHot measured 0 stars in 24 hours, 0 in 7 days, and 0 in 15 days. Last checked 2026-11-12 09:23 UTC. Last push 2024-02-18.
Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"
Measured star growth
- in 24h
- 0
- in 7 days
- 0
- in 15 days
- 0
- in 1 month
- Collecting history
Net star changes over overlapping periods, not daily totals. Missing measurements are left blank. Stars measure attention, not software quality.
README.md
OriginalClass-balanced-loss-pytorch
Pytorch implementation of the paper Class-Balanced Loss Based on Effective Number of Samples presented at CVPR'19.
Yin Cui, Menglin Jia, Tsung-Yi Lin(Google Brain), Yang Song(Google), Serge Belongie
Dependencies
- Python (>=3.6)
- Pytorch (>=1.2.0)
Review article of the paper
How it works
It works on the principle of calculating effective number of samples for all classes which is defined as:

Thus, the loss function is defined as:

Visualisation for effective number of samples

References
Explore related projects
tensorflow/tensorflow
An Open Source Machine Learning Framework for Everyone
201,505 starshuggingface/transformers
🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
168,815 starsAUTOMATIC1111/stable-diffusion-webui
Stable Diffusion web UI
165,300 starsComfy-Org/ComfyUI
The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface. The fastest local inference engine in the world.
146,472 stars