← all repositories

richardaecn/class-balanced-loss

TensorFlow implementation of class-balanced loss functions for handling class imbalance in fine-grained image classification.

615 stars Python Computer VisionML Frameworks
class-balanced-loss
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

This repository implements the CVPR 2019 paper on class-balanced loss based on effective number of samples, designed to address class imbalance in deep learning training. The code provides loss functions and training scripts for CIFAR datasets (both original and long-tailed variants), along with focal loss implementation. It targets fine-grained visual categorization problems using TensorFlow, supporting datasets like ImageNet and iNaturalist.

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