手把手教你运行yolov6 (小白版教程)

#一、 yolov6的介绍

#二、yolov6的吐槽

#三、yolov6运行详解

前言

自己运行yolov6的完整教程

#一、 yolov6的介绍
我在此应用美团视觉部官方的说法叙述一下yolov6
YOLOv6 是美团视觉智能部研发的一款目标检测框架,致力于工业应用。本框架同时专注于检测的精度和推理效率,在工业界常用的尺寸模型中:YOLOv6-nano 在 COCO 上精度可达 35.0% AP,在 T4 上推理速度可达 1242 FPS;YOLOv6-s 在 COCO 上精度可达 43.1% AP,在 T4 上推理速度可达 520 FPS。在部署方面,YOLOv6 支持 GPU(TensorRT)、CPU(OPENVINO)、ARM(MNN、TNN、NCNN)等不同平台的部署,极大地简化工程部署时的适配工作。
YOLOv6 是美团视觉智能部研发的一款目标检测框架,致力于工业应用。本框架同时专注于检测的精度和推理效率,在工业界常用的尺寸模型中:YOLOv6-nano 在 COCO 上精度可达 35.0% AP,在 T4 上推理速度可达 1242 FPS;YOLOv6-s 在 COCO 上精度可达 43.1% AP,在 T4 上推理速度可达 520 FPS。在部署方面,YOLOv6 支持 GPU(TensorRT)、CPU(OPENVINO)、ARM(MNN、TNN、NCNN)等不同平台的部署,极大地简化工程部署时的适配工作。在 COCO 数据集上,YOLOv6 在精度和速度方面均超越其他同体量算法,相关结果如下图所示:
手把手教你运行yolov6 (小白版教程)_第1张图片
YOLOv6 各尺寸模型与其他模型性能对比
手把手教你运行yolov6 (小白版教程)_第2张图片
Hardware-friendly 的骨干网络设计
YOLOv5/YOLOX 使用的 Backbone 和 Neck 都基于 CSPNet[5] 搭建,采用了多分支的方式和残差结构。对于 GPU 等硬件来说,这种结构会一定程度上增加延时,同时减小内存带宽利用率。下图 2 为计算机体系结构领域中的 Roofline Model[8] 介绍图,显示了硬件中计算能力和内存带宽之间的关联关系。
手把手教你运行yolov6 (小白版教程)_第3张图片

基于硬件感知神经网络设计的思想,对 Backbone 和 Neck 进行了重新设计和优化。上述重新设计的两个检测部件,我们在 YOLOv6 中分别称为 EfficientRep Backbone 和 Rep-PAN Neck,其主要贡献点在于:\n\n引入了 RepVGG[4] style 结构。\n\n基于硬件感知思想重新设计了 Backbone 和 Neck。\n\nRepVGG[4] Style 结构是一种在训练时具有多分支拓扑,而在实际部署时可以等效融合为单个 3x3 卷积的一种可重参数化的结构(融合过程如下图 3 所示)。通过融合成的 3x3 卷积结构,可以有效利用计算密集型硬件计算能力(比如 GPU),同时也可获得 GPU/CPU 上已经高度优化的 NVIDIA cuDNN 和 Intel MKL 编译框架的帮助。

Backbone部分,提出了一种叫做 EfficientRep Backbone 的结构,结构图如下:

手把手教你运行yolov6 (小白版教程)_第4张图片
图里的RepConv,RepBlock,SimSPPF均为全新的结构,这里不作更细的探究。

在 Neck 设计方面,YOLOv6提出了一种命名为Rep-PAN的结构,结构示意图如下:
手把手教你运行yolov6 (小白版教程)_第5张图片

Decoupled Head
在检测头方面,YOLOv6借鉴了YOLOX的思路,采用了解耦检测头(Decoupled Head)结构,并对其进行精简。两者对比图如下:
手把手教你运行yolov6 (小白版教程)_第6张图片

训练方法的改进
Anchor-free 无锚范式
YOLOv6 采用了更简洁的 Anchor-free 检测方法。检测方法。由于 Anchor-based检测器需要在训练之前进行聚类分析以确定最佳 Anchor 集合,这会一定程度提高检测器的复杂度;同时,在一些边缘端的应用中,需要在硬件之间搬运大量检测结果的步骤,也会带来额外的延时。而 Anchor-free 无锚范式因其泛化能力强,解码逻辑更简单,在近几年中应用比较广泛。
SimOTA
为了获得更多高质量的正样本,YOLOv6 引入了 SimOTA 算法动态分配正样本,进一步提高检测精度。YOLOv5 的标签分配策略是基于 Shape 匹配,并通过跨网格匹配策略增加正样本数量,从而使得网络快速收敛,但是该方法属于静态分配方法,并不会随着网络训练的过程而调整。
SIoU
使用了一种全新的Iou_Loss,近年来,常用的边界框回归损失包括IoU、GIoU、CIoU、DIoU loss等等,这些损失函数通过考虑预测框与目标框之前的重叠程度、中心点距离、纵横比等因素来衡量两者之间的差距,从而指导网络最小化损失以提升回归
精度,但是这些方法都没有考虑到预测框与目标框之间方向的匹配性。SIoU 损失函数通过引入了所需回归之间的向量角度,重新定义了距离损失,有效降低了回归的自由度,加快网络收敛,进一步提升了回归精度。通过在 YOLOv6s 上采用 SIoU loss 进行实验,对比 CIoU loss,平均检测精度提升 0.3% AP。

#二、yolov6的吐槽
首先是网友们的吐槽:
手把手教你运行yolov6 (小白版教程)_第7张图片
手把手教你运行yolov6 (小白版教程)_第8张图片
在这里插入图片描述
我只能说美团视觉部不自量力,瞎搞,代码的BUG特别多,在yolov6出来不久之后yolov4的作者便发布了yolov7,于是yolov6就成了一个笑话,美团现在也没发出一篇yolov6的论文,这东西拿来玩玩,做做对比实验吧。

#三、yolov6运行详解
准备工作

  1. GPU的准备
    我们需要租用一块GPU用来跑yolov6,这里推荐使用AutoDL平台,租用价格大约为100元。网址如下:https://www.autodl.com/home
    我们需要租用的GPU型号为

在这里插入图片描述
手把手教你运行yolov6 (小白版教程)_第9张图片
2. 数据集的准备
这里使用的是口罩数据集,数据集格式如下:
手把手教你运行yolov6 (小白版教程)_第10张图片
train中包含images和labels文件夹images包含图片,labels包含标签,val中同样也是。
3,。 pycharm的配置
用pycharm打开yolov6的工程
手把手教你运行yolov6 (小白版教程)_第11张图片
远程连接服务器,在选择解释器里点击添加,选择ssh,在AutoDL平台上的登录指令格式如下ssh -p 端口号@XXXXX
手把手教你运行yolov6 (小白版教程)_第12张图片
主机:@后面的XXX
用户名:root
端口号:-p后面的数字
接下来输入密码
手把手教你运行yolov6 (小白版教程)_第13张图片
连接服务器后,找到root/minconda3/python作为解释器,不要Python3
手把手教你运行yolov6 (小白版教程)_第14张图片
点击完成应用,在工具中点击自动上传,修改myself.yaml文件
手把手教你运行yolov6 (小白版教程)_第15张图片
将names:改成[‘mask’,‘no-mask’],接下来上传权重和数据集,这里使用的是filezilla,使用方法有很多

手把手教你运行yolov6 (小白版教程)_第16张图片
在上传完成之后,我们需要更改myself.yaml中的路径地址换成终端里的地址

手把手教你运行yolov6 (小白版教程)_第17张图片
出现报错,不知道怎么办从来,然后发现将train.py和infer.py单独上传
手把手教你运行yolov6 (小白版教程)_第18张图片

手把手教你运行yolov6 (小白版教程)_第19张图片

手把手教你运行yolov6 (小白版教程)_第20张图片

接下来会缺少包,我们需要再远程终端中下载

手把手教你运行yolov6 (小白版教程)_第21张图片
启动SSH会话进入终端,下载如下包。
在这里插入图片描述
手把手教你运行yolov6 (小白版教程)_第22张图片
手把手教你运行yolov6 (小白版教程)_第23张图片

手把手教你运行yolov6 (小白版教程)_第24张图片
接下来出现如下错误
手把手教你运行yolov6 (小白版教程)_第25张图片

原因我很无语,美团我就不问候他了
手把手教你运行yolov6 (小白版教程)_第26张图片
这里改成yolov6,然后我们进行训练显示找不到数据集,经过查找文档发现yolov6格式不同于yolov5和yolov7,我们修改后重新上传。
手把手教你运行yolov6 (小白版教程)_第27张图片
v6不需要使用images文件夹,将以前images中的train和val文件夹图片与labels文件放在同一个目录就可以了

接下来运行train.py,运行成功后
手把手教你运行yolov6 (小白版教程)_第28张图片
找到best_ckpt.pt权重替换原来的yolov6s.pt就可以进行训练了!
手把手教你运行yolov6 (小白版教程)_第29张图片

还有一些常见错误,大家可以看这篇博客。
http://t.csdn.cn/sApoW
特别感谢对卿云阁我的帮助
大家多多关注,打赏他。
手把手教你运行yolov6 (小白版教程)_第30张图片

你可能感兴趣的:(人工智能,深度学习)