基于Yolov5的火灾检测系统_专栏

文章目录

一、YOLOv5介绍
二、模型详解
三、项目意义
四、检测效果
五、数据集获取
六、模型训练效果
七、总结

一、YOLOv5介绍

YOLO(You Only Look Once) 是REDMON等提出的目标检测算法，作为单阶段(one-stage)的代表算法，目前 YOLO已经更新到了YOLOv5。相比于Two stage目标检测算法，YOLO是直接从网络中提取特征，并预测物体的类别和具体位置，一步到位。

此后，REDMON在此基础上提出了YOLO9000等检测算法，使系统的检测性能得到进一步提升。在YOLOv3的基础上继续改进升级，并最终得到 YOLOv4。YOLOv4网络结构主要由CSPDarknet53特征提取网、SPP(空间金字塔池化)模块、PANet特征融合模块、Yolo Head分类器组成。

YOLOv5算法使用CSPDarknet(跨阶段局部网络)作为特征提取网络,从输入图像中提取目标信息。如今，YOLOv5无论是在准确率还是速度上，都已经达到较好的效果。所以，本项目采用YOLOv5训练模型，结合YOLOv5算法构建火灾检测系统。

二、模型详解

上图是yolov5s的网络结构，它是yolov5系列中深度最小、特征图宽度最小的网络。后面的m、l、x都是在此基础上不断加深、加宽的。网络主要分为输入端、Backbone、Neck、Prediction四个部分。它和yolov3主要不同的地方：

输入端：Mosaic数据增强、自适应锚框计算、自适应图片缩放
Backbone：Focus结构、CSP结构
Neck：FPN+PAN结构
Prediction：GIOU_Loss
YOLOv5 模型在输入端(Input)增加了 Mosaic 数据增强、自适应锚框计算、自适应图片缩放等数据预处理技巧来增强数据，防止过拟合；在特征提取网络部分(Backbone)引入了Focus 模块、跨阶段局部融合网络(CrossStage Partial Network, CSPNet)等方法，在减少了计算量的同时可以保证准确率，使特征能够更好的向后传递。下图为cspnet 结构图：

CSPNet主要目的就是缓解以前需要大量推理计算的问题，它有以下优点：

增强了CNN的学习能力，能够在轻量化的同时保持准确性。
降低计算瓶颈。
降低内存成本。

CSPNet通过将梯度的变化从头到尾地集成到特征图中，在减少了计算量的同时可以保证准确率。

CSPNet和PRN都是一个思想，将feature map拆成两个部分，一部分进行卷积操作，另一部分和上一部分卷积操作的结果进行concate。在特征融合部分(Neck)借鉴了空间金字塔池化(Spatial Pyramid Pooling, SPP)、特征金字塔(Feature Pyramid Networks, FPN)与路径聚合网络(PathAggregation Network, PAN)等方法，有效的增加主干特征的接收范围，融合后的特征保留了丰富的语义特征和精准的定位特征；同时在预测部分(Prediction)采用了新的定位损失函数，确保定位的精准。通过引入各种高效的网络组件，使得YOLOv5 模型在保持实时目标检测速度的基础上，也实现了精度上SOTA。

2.1 Input

Input 使用了 Mosaic 数据增强、自适应锚框计算、图片尺寸处理。Mosaic 数据增强把 4 张图片，采用随机缩放、随机裁剪、随机排布的方式进行拼接 , 极大地丰富了检测数据集，同时也能提高小目标检测的精度，除此以外，Mosaic数据增强在训练时可一次性对 4 张图片的数据进行计算，降低了 mini-batch 的大小，也减少了 GPU 的使用。YOLOv5 将自适应锚框计算嵌入代码中，通过在初始设定的锚框上输出预测框，然后和真实标签进行比对，计算损失函数，再不断更新，更新锚点框的大小，实现自适应计算并得出最佳锚框值。图片尺寸处理是对输入的各种图片的尺寸进行自适应填充处理，为了提高目标检测的速度，采用了减少灰度边缘的方法。

2.2 Backbone

Backbone 是 YOLOv5 网络的主干部分，包含 Focus 结构、CSP 结构和 SPP 结构。其中 Focus 结构：主要进行切片操作，只存在于YOLOv5 算法中，以 YOLOv5s 为例，将原始三通道图像输入 Focus 结构，经过切片操作后，进行拼接，图片尺寸缩小到原来的 1/4、输入通道扩充到原来的 4 倍，经过 32 个卷积核的卷积计算，最终得到含有 32 个通道的特征图。YOLOv5s、YOLOv5m、YOLOv5x 和 YOLOv5l，使用的卷积核数量依次增加，其中，YOLOv5m 使用了 48 个卷积核。卷积核个数越多 , 特征图的宽度越宽 , 网络提取特征的学习能力也越强。开发者认为，Focus 模块的设计目的是减少层数并降低计算量。

CSP 结构：YOLOv5 中设计了 CSP1_N 和 CSP2_N 两种CSP 结构，CSP1_N 应用于 Backbone 作为主干网络，CSP2_N 应用在 Neck 中，这部分没有残差组件。CSP1_N 结构将基础层的特征映射划分为两个不同的部分，其中之一就是将 N个残差组件进行卷积操作，另一部分则是直接进行卷积操作，两次卷积操作可以使通道数减半，然后通过拼接来进行输出。
SPP 结构：在 Backbone 中，采用 SPP（空间向量金字塔池化）。多尺度融合是通过最大化池来实现的。

2.3 Neck

Neck 采用 FPN+PAN 的结构。FPN 结构是自上向下传递强特征，对结构起到增强的作用，然而，FPN 只能增强语义信息，而不能传递位置信息。而 PAN 结构刚好弥补了 FPN结构不能进行定位信息传递的缺陷，PAN 结构自下向上将低层的强定位特征传递上去，两者结合操作，增强网络特征融合的能力。

2.4 Prediction

本文中 Yolov5 采用 GIOU_Loss 作 Bounding box 的损失函数GIOU_Loss 能够区分重叠的预测框和目标框在 IOU 相同时两者相交情况的不同。另外，针对预测过程中会出现多个预测框的情况，需要通过非极大值抑制即 NMS 来处理，Yolov5 使用加权 NMS 来筛选最佳的预测框

三、项目意义

火灾作为威胁人类生命生产安全的隐患之一，一直是人们关注的重点。传统的火灾监测装置根据温度来检测火灾，不仅灵敏度差，而且反馈时间长，常常会出现消防员收到警报消息时，火室已经无法控制。本文由树莓派、Opencv和蜂鸣器设计了一套火灾检测装置，能够对火灾进行实时灵敏检测，适用于地下停车场、居民楼道、商场等多种场景。

四、检测效果

五、数据集获取

通过labelImg 标注工具对数据集图像进行标注,并保存为YOLO 系列的 txt 格式。可以在你的虚拟环境中进入labelImg，这是他的界面。左侧open Dir可以打开数据集的文件夹，Change Save Dir是你的保存路径，Next和Prev Image分别是上一张和下一张图片。重点来了：Create RectBox绘制一个矩形框将你需要框选的对象框选出来，并添加标签。这里不懂的同学可以看我之前有一篇很详细的教程链接: Yolov5：强大到你难以想象──新冠疫情下的口罩检测

通过对数据进行乱序排列,随机选出训练集 1442 张,测试集共617 张,验证集共 617 张。通过上千张火灾的图片进行机器学习的训练，对于微小火焰也有这不错的检测效果。数据集的下载地址我贴在这里了：链接：https://pan.baidu.com/s/1ry7o3oJfyHM5rw4nkXL14A?pwd=f5xp
提取码：f5xp