search

hukaixuan19970627 / yolov5_obb

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

训练时默认rect参数为False，可以更改为true吗？ #269

Closed kurt369 closed 2 years ago

kurt369 commented 2 years ago

不开启矩形训练感觉速度还是很慢，开启后会影响旋转目标的预测框的定位吗？

hukaixuan19970627 commented 2 years ago

首先你要分析限制训练速度提升的原因是什么。例如：查看训练时的gpu利用率，如果时常利用率为0，说明一轮数据的训练完成之后下一轮的数据还没准备好，这表明电脑的cpu性能应付不了这么多的数据量，这个时候减小batchsize反而能提升整体的训练效率。

hukaixuan19970627 commented 2 years ago

另外关于矩形训练的rect参数，只要trainbatch0/1/2/.../n.jpg里的labels没问题那就随你更改

kurt369 commented 2 years ago

谢谢大哥回答，另外还有两个问题:  1）专用Gpu内存利用率基本是满的（21.2/24），copy那一项在训练时一直是0，在验证的时候就跳到88%，这是正常的吗？ 2）我是40002000的图片，imgsize设置的是4000，按照yolo的处理方式，没开启矩形训练的话，会被pad成40004000，在开启矩形训练之后，应该会被pad成4000*2016，按理说开启矩形训练之后，gpu占用和训练时间都会减小才对，可是我开启矩形训练之后，发现gpu占用和训练时间并没有变化，这是什么原因呢？

---原始邮件--- 发件人: @.> 发送时间: 2022年4月8日(周五) 下午2:26 收件人: @.>; 抄送: @.**@.>; 主题: Re: [hukaixuan19970627/yolov5_obb] 训练时默认rect参数为False，可以更改为true吗？ (Issue #269)

另外关于矩形训练的rect参数，只要trainbatch0/1/2/.../n.jpg里的labels没问题那就随你更改

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

hukaixuan19970627 commented 2 years ago
  1. gpu利用率（Volatile GPU-Util）和gpu占用率（Memory-Usage）是两个概念。
  2. 经测试840×712分辨率的图像在使用矩形训练之后可降低30%左右的显存消耗量。