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。但谷歌colab使用的是相同的代码

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

矩阵混乱，result.csv几乎为空，还有nan文本

谢谢你的帮助

Answer 1

我对YOLOv5也有类似的问题，通过关注 img（图像大小）和 conf属性，我能够解决这个问题

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

python train.py  img 672 1216 ...

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

python detect.py  img 672 ...

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

python detect.py ...  conf 0.7

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