Skip to content
Projects

vandit15/Class-balanced-loss-pytorch

View on GitHub

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.

How these counts are measured

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

Original

Class-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

Medium Article

How it works

It works on the principle of calculating effective number of samples for all classes which is defined as:

alt-text

Thus, the loss function is defined as:

alt-text

Visualisation for effective number of samples

alt-text

References

official tensorflow implementation

Explore related projects