search

DataXujing / YOLO-v5

:art: Pytorch YOLO v5 训练自己的数据集超详细教程！！！ :art: (提供PDF训练教程下载）
GNU General Public License v3.0
913 stars 196 forks source link

Hi，您好，我想问一下关于数据不均衡问题 #26

Open young169 opened 3 years ago

young169 commented 3 years ago

需要关注吗？不均衡的数据mAP只有0.2左右。根据您的经验建议怎么解决呢？是在数据准备中还是在loss中加权？

DataXujing commented 3 years ago

做一些离线的数据增强，或替换Loss, 现在YOLO v5使用Focal Loss, 可以试一下Gensralized Focal Loss等； 但是最重要的还是数据量和标注的质量，这两个肯定要保证，如果数据量上来了，不平衡也会被削弱

young169 commented 3 years ago

感谢！不过还想问一下怎么样具体的显示验证集中每一类的precision、recall呢？还有就是怎么样固定部分网络的权重，比如我只想训练用于分类的网络？