How can I use this method in yolov5?

megvii-research / BBN

The official PyTorch implementation of paper BBN: Bilateral-Branch Network with Cumulative Learning for Long-Tailed Visual Recognition

https://arxiv.org/abs/1912.02413

MIT License

657 stars 100 forks source link

How can I use this method in yolov5? #22

Open JiangWeiHn opened 3 years ago

JiangWeiHn commented 3 years ago

I want to train my dataset with yolov5, but my dataset has a long tail problem. Some categories have little data. Can I use your method to solve this problem? if can't,please give me some advice，thanks a lot