Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
young169 opened this issue Jan 27, 2021 · 2 comments
Open

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

young169 opened this issue Jan 27, 2021 · 2 comments

Comments

@young169
Copy link

young169 commented Jan 27, 2021

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

@DataXujing
Copy link
Owner

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

@young169
Copy link
Author

young169 commented Mar 30, 2021

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants