hukaixuan19970627 / yolov5_obb

yolov5 + csl_label.(Oriented Object Detection)(Rotation Detection)(Rotated BBox)基于yolov5的旋转目标检测
GNU General Public License v3.0
1.81k stars 425 forks source link

训练自己的数据集,拟合并不理想 #82

Closed zhaoguoqing12 closed 2 years ago

zhaoguoqing12 commented 3 years ago

image train_batch0 train_batch1 train_batch2

测试自己的数据不是很理想,帮忙看一下吧 1.似乎box loss 以及 angle loss并没有收敛,两分类,只不过数据量差距较大。 labels 2.超参数并没有修改,训练了80周期loss不下降啦,数据集在2000张左右,效果远远没有达到预期。

github-actions[bot] commented 3 years ago

Hello @zhaoguoqing12, thank you for your interest in our work! Please visit our Custom Training Tutorial to get started, and see our Jupyter Notebook Open In Colab, Docker Image, and Google Cloud Quickstart Guide for example environments.

If this is a bug report, please provide screenshots and minimum viable code to reproduce your issue, otherwise we can not help you.

If this is a custom model or data training question, please note Ultralytics does not provide free personal support. As a leader in vision ML and AI, we do offer professional consulting, from simple expert advice up to delivery of fully customized, end-to-end production solutions for our clients, such as:

For more information please visit https://www.ultralytics.com.

hukaixuan19970627 commented 3 years ago

接着训练完的权重继续训练。 mosaic关闭,水平翻转关闭,剩下翻转关闭,你的场景不适合用这些数据增强策略

zhaoguoqing12 commented 3 years ago

接着训练完的权重继续训练。 mosaic关闭,水平翻转关闭,剩下翻转关闭,你的场景不适合用这些数据增强策略 谢谢 我会试试的