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

关于hrsc2016数据集 #201

Closed shenhaibb closed 2 years ago

shenhaibb commented 2 years ago

我用HRSC2DOTA.py进行转换数据集

image

进行训练的时候 QQ20220214-213011@2x 然后跑到一半就会报错 QQ20220214-213025@2x 为什么转换后的数据集会有负值呢

shenhaibb commented 2 years ago

出现warning的图片,cache会排除这些图片进行训练吗

shenhaibb commented 2 years ago

https://github.com/hukaixuan19970627/yolov5_obb/issues/147#issuecomment-1009625752 我试试这个

hukaixuan19970627 commented 2 years ago

自定义数据集在转成poly形式坐标时,做一下clip处理,使坐标取值范围落在图像范围内。 如果自定义数据集包含切割操作的话,参考一下dota数据集对待不完整目标的做法(最好是先转成dota形式然后再用dota_devkit这样最方便): image

另外srcIndex那个问题在最新提交的PR https://github.com/hukaixuan19970627/yolov5_obb/commit/0fd0d673eac4f2478822d5d526f237c2eaf49123 中已修复