ShadeAlsha/LTR-weight-balancing

CVPR 2022 - official implementation for "Long-Tailed Recognition via Weight Balancing" https://arxiv.org/abs/2203.14197

37
/ 100
Emerging

When training image recognition models, you often find that the model performs poorly on categories with less training data, while doing very well on categories with a lot of data. This project helps you train more balanced models that are better at recognizing objects from 'rare' categories. It takes your existing image dataset and training setup, and outputs a more accurate, balanced image recognition model. This is useful for anyone building computer vision systems where some object categories appear less frequently than others.

128 stars. No commits in the last 6 months.

Use this if your image recognition models struggle to identify objects from categories that have fewer examples in your training data.

Not ideal if your dataset has a perfectly even distribution of images across all object categories, or if you are not working with image recognition.

image-recognition computer-vision machine-learning-training object-detection imbalanced-data
Stale 6m No Package No Dependents
Maintenance 0 / 25
Adoption 10 / 25
Maturity 16 / 25
Community 11 / 25

How are scores calculated?

Stars

128

Forks

11

Language

Jupyter Notebook

License

MIT

Last pushed

Nov 30, 2024

Commits (30d)

0

Get this data via API

curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/ShadeAlsha/LTR-weight-balancing"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.