YOLOv5-P6

系列文章

 

• YOLOv5 环境配置和安装

训练报错

assertionError: 1 names found for nc=15 dataset in .yaml

训练的时候注意配置文件中的类数和类名的数量要对应，不然会报assertionError

AssertionError: Label class 15 exceeds nc=1 in data/coco128.yaml. Possible class labels are 0-0

标签.txt文件标注从'0'开始

Results.txt

results.txt里后面，yolov5生成的结果数据有数据没属性，最后几列是何一直不确定，github上也没有解释，改了results.txt的数据，画图做了个小实验才确定下来。

>>> from utils.plots import plot_results
>>> plot_results(save_dir="./yolov5-develop/train/exp/')
#自动读取路径下的results.txt文件，并生成results.png

1. epoch
2. gpu_mem
3. box: 推测为GIoU损失函数均值，越小方框越准；
4. obj: 推测为目标检测loss均值，越小目标检测越准