YOLOv5-P6

本文介绍了YOLOv5训练中遇到的assertionError,强调了配置文件中类数与类名的一致性。讨论了results.txt的各列含义,特别是与损失函数和性能指标相关的部分。深入探讨了[email protected]的计算方法,包括查准率和查全率的概念,并解释了PR曲线的含义。

系列文章

 

训练报错

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均值,越小目标检测越准
  5. classification: 推测为分类loss均值,越小分类越准;
  6. total
  7. lab
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值