PaddlePaddle / PaddleYOLO

🚀🚀🚀 YOLO series of PaddlePaddle implementation, PP-YOLOE+, RT-DETR, YOLOv5, YOLOv6, YOLOv7, YOLOv8, YOLOv10, YOLOX, YOLOv5u, YOLOv7u, YOLOv6Lite, RTMDet and so on. 🚀🚀🚀
https://github.com/PaddlePaddle/PaddleYOLO
GNU General Public License v3.0
547 stars 133 forks source link

tt100k yolov5up6 训练报错 #159

Open hbo-lambda opened 1 year ago

hbo-lambda commented 1 year ago

问题确认 Search before asking

请提出你的问题 Please ask your question

ubuntu 20.04 paddle 2.4.2 nvidia 单卡 v100 32G

执行命令:

python tools/train.py -c configs/yolov5u/yolov5up6_s_300e_coco.yml --eval --amp

调整了数据集的样本数量,使用roadsign_voc,样本数量少于32的时候会报错。所以自己选取原样本中35个样本去测试,我的标签数量是250个,抽取的样本尺寸是2048*2048。yolov5up6_s_300e_coco.yml 相关的配置,除了数据集路径外,其他没做改动。抽取的样本如链接: https://pan.baidu.com/s/1TvUUjSGGShppZAnewPURPA?pwd=cnen 提取码: cnen

报错如下: Snip20230615_5

nemonameless commented 1 year ago

请先尝试yolov5s或别的模型,如果也报同样的错,则可认为是数据集制作的问题。

mutouoa commented 1 year ago

@nemonameless 出现同样的问题,也已在其他模型训练过

nemonameless commented 6 months ago

你好,如还需要帮看下问题的话还请提供下您的数据集。库里的模型是在coco voc上验证过的,应该是大部分数据集边界情况都覆盖了。