YOLOv8在客制化数据集上训练出现SystemError

LorenzoSun-V commented 9 months ago

问题确认 Search before asking

[X] 我已经查询历史issue，没有发现相似的bug。I have searched the issues and found no similar bug report.

Bug组件 Bug Component

Training

Bug描述 Describe the Bug

我制作了一个客制化数据集，用labelimg标注后转成coco格式，其中有一些负样本没有标签，在coco标签中的表现为在images字段有image_id，但在annotations字段没有对应image_id的标签。在数据集中，我已设置allow_empty=True，并在yolov5和yolox上可以正常训练和推理。但在yolov8训练时，会出现以下bug：看报错信息的话，dataset已经加载成功，但是在Dataloader取数据的时候报错了。我把worker_num设置为0，可以正常训练，但是loss太大就卡住了：这个问题我在训练yolov7u的时候也遇到了。

我的配置文件为：修改的optimizer_100e.yml为：

复现环境 Environment

OS: Linux
PaddlePaddle-gpu: 2.5.1.post117
PaddleDetection: 2.6.0
Python: 3.7.14
CUDA: 11.7
GCC: 8.2.0

Bug描述确认 Bug description confirmation

[X] 我确认已经提供了Bug复现步骤、代码改动说明、以及环境信息，确认问题是可以复现的。I confirm that the bug replication steps, code change instructions, and environment information have been provided, and the problem can be reproduced.

是否愿意提交PR？ Are you willing to submit a PR?

[ ] 我愿意提交PR！I'd like to help by submitting a PR!

LorenzoSun-V commented 9 months ago

另外我试了一下原始的coco配置文件训练，yolov8是可以正常训练的。

LorenzoSun-V commented 9 months ago

根据 #164 的回答，把 use_shared_memory 设为False可以正常训练，但是同时也出现了训练loss一直为0的情况。

nemonameless commented 4 months ago

https://github.com/PaddlePaddle/PaddleYOLO/issues/166 谢谢建议，后续进展在166中更新

PaddlePaddle / PaddleYOLO