yolov5不训练模型

Question

我用yolov5训练了一个数据集。模型是存在的，但它不起作用。它不会过度预测。

下面是我使用的代码：

python train.py --img 640 --batch 8 --epochs 3 --data ../datasets/coco128/coco128.yaml --workers 0 --image-weights '' --name test

代码的输出：

​

​

​

​

​

​

当我们查看用于预测的测试文件夹时：

标签

​

​

预测

​

​

如你所见，预测什么都不是。还有best.pt和last.pt文件。它们的大小是14.475 mb。但是google colab使用相同的代码。

此外，当我尝试用默认的yolov5s.pt预测图像时，它是有效的，预测是正确的。

矩阵混淆，带有nan文本的result.csv几乎是空的。

谢谢你的帮助...

Answer 1

我在使用YOLOv5时遇到了类似的问题，并通过注意--img (图像大小)和--conf属性解决了这个问题。

调用训练脚本时，您可以定义图像尺寸

python train.py --img 672 1216 ...

但是您可能忘记了在推理脚本中使用相同的参数

python detect.py --img 672 ...

另一个问题是当你的数据集太小，召回率/精确度/mAP太低时，你需要在推理过程中减少输出的置信度过滤器：

python detect.py ... --conf 0.7

您可能需要将--conf减小到0.1才能看到任何输出。然后使用它，直到您对输出中的精度/召回率感到满意。