YOLOv5训练自己的模型时，出现Traceback (most recent call last)的问题

1 问题描述

用YOLOv5用detect.py用下述指令查看训练效果的时候，出现如下图所示的问题。

python detect.py --source data/images/test --weight mytrain.pt

2 尝试解决

先用其他权重文件排除了数据图像的问题。

发现在训练过程中，，因为在网上下载的数据集，配置yaml文件时指定数据的类别，我之前写的是1，但是数据集标注的类型序号通过训练过程中的图片显示是0不匹配，可能是因为该原因出现的问题。

更改yaml文件names,将其设置成0，再重新训练即可。

3 数据标注的思考

在网上拿到的数据集，对于新手来说，不知道数据标签的意思。其实最前面的一列表示的是检测到的数据类别编号，第二个和第三个数据分别代表目标框的中心元素的坐标占图片大小的百分比，最后两个数据是目标框的长和宽所占整个图像像素的比。

4 总结

  大家在训练YOLOv5之前，一定要改好.yaml文件，否则就会出现训练好的权重文件不可用的问题。