基于YOLOv8化工企业储煤场火煤和扬尘检测方法研究.pdf

盐科学与化工Journal of Salt Science and Chemical Industry第53卷第3期2024年3月收稿日期：2023-11-15作者简介：禹晨（1991），男，硕士研究生在读，研究方向为深度学习、图象处理。联系方式：18547211298，基于YOLOv8化工企业储煤场火煤和扬尘检测方法研究禹晨，张飞，郝斌（内蒙古科技大学 信息工程学院，内蒙古 包头014000）摘要：化工企业储煤场的火煤和扬尘不仅浪费了煤炭资源，还污染了周围环境，对工作人员的健康构成威胁。为了减少经济损失、改善环境，对火煤和扬尘的监测和预警变得极为关键。传统的储煤场火煤和扬尘的检测存在检测时间长、误检率高等问题。文章在深入研究YOLOv8、SPD-Conv、BottleNeckCSP、EIoU原理的基础上，对YOLOv8模型进行改进。用改进后的YOLOv8网络模型完成火煤和扬尘检测相关实验，与原始的YOLOv8相比，mAP值达到97.3%，比YOLOv8s原始模型提高了3.9%，优于同等情况下YOLO系列，且权重文件的内存为25.4 MB，适用于火煤和扬尘的检测，达到对于低分辨率和小目标更快、更精准的检测，有效避免误检、漏检等问题，确保检测的准确性和可靠性。关键词：火煤和扬尘检测；YOLOv8模型改进；低分辨率和小目标检测中图分类号：X84；TP391.41文献标识码：A文章编号：2096-3408（2024）03-0049-06Research on the Detection Method of Fire Coal andDust in Coal Storage Yard of YOLOv8Chemical EnterpriseYU Chen，ZHANG Fei，HAO Bin（School of Information Engineering，Inner Mongolia University of Science andTechnology，Baotou 014000，China）Abstract：The hot coal and dust in the coal storage yard of chemical enterprises not onlywaste coal resources，but also pollute the surrounding environment，posing a threat to the health ofworkers.In order to reduce economic losses and improve the environment，monitoring and earlywarning of coal and dust have become increasingly crucial.The traditional detection of hot coal anddust in coal storage yards has problems such as long detection time and high false detection rate.On the basis of in-depth research on the principles of YOLOv8，SPD-Conv，BottleNeckCSP，andEIoU，this article improves the YOLOv8 model.The improved YOLOv8 network model was used tocomplete experiments related to the detection of coal and dust.Compared with the original YOLOv8，the mAP value reached 97.3%，which is 3.9%higher than the original YOLOv8s model andbetter than the YOLO series under the same conditions.The weight file has the size of 25.4 MB，which is suitable for the detection of coal and dust，achieving faster and more accurate detection oflow resolution and small targets，effectively avoiding problems such as false positives and misseddetections，Ensure the accuracy and reliability of testing.Key words：Fire coal and dust detection；YOLOv8 model improvement；Low resolutionand small object detection49盐科学与化工第53卷第3期2024年3月当煤与空气接触并达到其燃点时，它会发生局部自燃，这不仅导致煤炭资源的浪费，还会向大气中排放大量的一氧化碳、二氧化硫和其他有害气体，对环境造成污染。同时，运输和处理煤炭过程中产生的扬尘无法完全避免，这不仅会污染周围环境，还会对工作人员的健康产生危害。因此，为了降低经济损失、减少环境污染并改善工作环境质量，对火煤和扬尘的监测和预警变得至关重要。近年来，深度学习技术在目标识别领域已经有了广泛应用，目标检测的主要功能是检测图像中出现的既定目标类别以及目标的位置信息。学术界同时也涌现出许多基于烟尘的检测方法，主要分为两类：一类是以 Faster R-CNN1与 Mask R-CNN2等为代表的两阶段检测算法；另一类是以 YOLO3等为代表的一阶段目标检测算法。很多学者为提高检测精度和速度做了大量工作，文献 4 通过使用MobileNetv35重构SSD6的主干使模型轻量化；文献7 使用深度可分离卷积来替换YOLOv48网络中的普通卷积实现模型的轻量化。但是模型的轻量化往往会带来检测精度下降的问题，随着 YOLO系列的快速更迭，YOLOv8相对于其他目标检测算法，在保持高精度的同时，具有更快的推理速度，采用轻量级网络结构使其在移动设备上运行更加流畅。煤场环境下的火煤和扬尘检测，对模型的轻量化和检测精度都提出了很高的要求，而且煤炭自燃初期，产生的烟往往是通过小目标呈现的，煤场环境下的尘雾、灯光这些类烟雾等物体都会对检测造成干扰，所以提高复杂场景下对分辨率低图片中烟尘的小目标检测速度、精度是亟待解决的问题。文章针对以上问题，对YOLOv8模型进行改进，在主干特征提取网络 Backbone 层部分，引入 SPDConv代替原有的普通卷积，提高模型对于分辨率低的图像中烟尘和其呈现为小目标状态下的检测精度；在特征融合 Neck 层采用 BottleNeckCSP 模块替换之前的C2f模块，可以在主干特征提取阶段减少运算参数量，既保证了推理的速度和准确性，还减小了模型尺寸，使模型更加轻量化，提高了检测速度。除此之外，还改进YOLOv8损失函数由CIoU变为 EIoU，以达到使模型训练时收敛更加快速的目的，能够有效地描述回归任务，以便更精准地对火煤和扬尘进行检测。1YOLOv8算法YOLO 是一种经典的一阶段目标检测算法。YOLOv19提出了一个全新的目标检测框架，将目标检测任务转化为一个回归问题，通过一个单一的神经网络模型同时预测目标的位置和类别。YOLOv210在YOLOv1的基础上进行了改进，引入了anchor boxes，使得模型能够更好地处理不同大小的目标。同时，YOLOv2还采用了更复杂的损失函数，提高了模型的训练效果。YOLOv3、YOLOv4 在YOLOv2的基础上进一步改进，采用了更深的神经网络结构，提高了模型的性能，同时还引入了多尺度特征融合，使得模型能够更好地处理不同尺度的目标。YOLOv5采用了新的CSPNet骨干网络架构，计算效率和精确率都有所提升。相比于之前的YOLO系列，YOLOv8在训练时间和检测精度上都得到提升，而且模型的权重文件相比之前的YOLO系列进一步轻量化，可以部署到任一嵌入式设备中，它凭借自身快速、高效的性能可以很好地满足实时检测的需求。YOLOv8的网络结构由三个部分组成：主干网络（backbone）、特 征 增 强 网 络（neck）、检 测 头（head）。其中主干网络依然使用CSP的思想。而在特征增强网络方面，YOLOv8 使用 PAN-FPN 的思想。在检测头部分，YOLOv8使用解耦头替换了以往的耦合头11，将分类和回归解耦为两个独立的分支，通过解耦使各个任务更加专注，从而解决复杂场景下定位不准及分类错误的问题。同时YOLOv8采用了Anchor-free目标检测方法12，这是一种基于回归的目标检测方法，它不需要预先定义锚点框来预测目标的位置。在传统的目标检测方法中，通常需要预先定义锚点框，这些锚点框用于作为预测目标位置的参考。然而，这种方法存在一些问题，如锚点框的选择和调整比较繁琐，而且对于不同尺度和形状的目标，可能需要不同的锚点框。相比之下，YOLOv8的Anchor-free目标检测方法采用了一种回归的方式，直接预测目标的位置和大小，让网络更快地聚焦到目标位置的邻近点，使预测框更接近于实际边界框区域。2Smoke-YOLOv8s网络2.1SPD Conv模块SPD Conv 是一种用于替代跨步卷积与池化操作的特殊卷积，它由一个空间到深度（SPD）层和一个非50步长卷积（Conv）层组成。空间到深度（SPD）层的作用是将输入特征图的每个空间维度降低到通道维度，同时保留通道内的信息。这可以通过将输入特征图的每个像素或特征映射到一个通道来实现，SPD Conv 模块采样（见图1）。在这个过程中，空间维度的大小会减小，而通道维度的大小会增加。非步长卷积（Conv）层是一种标准的卷积操作，它在SPD层之后进行。与步长卷积不同，非步长卷积不会在特征图上移动，而是对每个像素或特征映射进行卷积操作。这有助于减少在SPD层中可能出现的过度下采样问题，并保留更多的细粒度信息。SPD Conv的组合方式是将SPD层和Conv层串联起来。具体来说，输入特征图首先通过SPD层进行转换，然后输出结果再通过Conv层进行卷积操作。这种组合方式可以在不丢失信息的情况下减少空间维度的尺寸，同时保留通道内的信息，有助于提高对低分辨率图像和小型物体的检测性能。在火煤自燃的初期，通常在产生火焰之前，首先会产生烟雾，但是初期烟雾在图象中所占面积比例很少，以小目标形式出现，而且图象质量受环境影响，很多时候用于检测的图像，分辨率并不高，引入SPD Conv模块代替主干特征提取网络中的普通卷积来提高分辨率低的图像中烟尘和其呈现为小目标状态下的检测精度。2.2BottleNeckCSP模块在网络的特征提取部分，采用了BottleNeckCSP模块来替换之前的C2f模块（见图2）。这一改进的目的是为了更好地学习和提取网络残差特征，同时调整特征图的深度和宽度。与原网络中的C2f模块相比，BottleNeckCSP模块能够减少内存消耗和计算瓶颈，提高网络的效率和性能。BottleNeck 结构通过卷积计算改变数据的通道数，从而调整特征图的深度和宽度。这种设计可以在主干特征提取阶段减少运算参数量，既保证了推理的速度和准确性，还减小了模型尺寸，使模型更加轻量化。2.3EIoU损失函数损失函数是判断模型预测值和预期值差距的一项重要指标，在一定程度上反映了模型对预测结果的准确度，预测值和预期值越接近，损失函数就越小。合适的损失函数可以使模型训练时收敛更加快速，所以在训练模型中选择合适的损失函数尤为重要13。在 YOLOv8中使用 CIoU 作为坐标损失函数，如公式（1）所示。LCIoU=1-IoU+2(b，bgt)c2+（1）式中：b为目标框的中心点；bgt为预测框的中心点；c 为目标框和预测框的最小外接框的对角线距离；为目标框和预测框中心点的欧氏距离；为权重参数。在原始YOLOv8中，CIoU损失函数在预测框回归过程中，当预测框和真实框宽高纵横比呈现线性比例时，预测框宽和高不能同时增加或减少，不能有效地描述回归任务，从而使整体损失函数收敛变慢14。文章根据原YOLOv8损失函数不足之处，采用EIoU损失函数替换YOLOv8中使用的CIoU损失函数。EIoU 损失函数主要是在 CIoU 的基础惩罚项上，将预测框和真实框的横纵比影响因子拆分后分别计算预测框和真实框的长、宽。EIoU 公式如公式（2）所示。LEIoU=LIoU+Ldis+Lasp=1-IoU+2(b，bgt)(wc)2+(hc)2+2(w，wgt)(wc)2+2(h，hgt)(hc)2式中：LIoU、Ldis、Lasp分别为IoU损失、中心距离损失、方位损失；wc、hc分别为最小闭合框的宽和高。CIoU和EIoU损失函数迭代过程见图3。其中，点线框代表真实框，黑实线框代表预先设定的锚框，点划线框代表CIoU的预测框回归过程，虚线框代表EIoU的预测框回归过程。从图3可以看出，使用CIoU时，预测框的宽和高不能同时增大或减小，EIoU很好地解决了这个难题。2.4构建火煤和扬尘检测网络文章在YOLOv8的基础上，针对以上内容进行改进设计，构建Smoke-YOLOv8s火煤和扬尘的检测网络，用于火煤和扬尘的检测实验研究见图4。3实验测定与结果分析3.1构建数据集从企业收集了6 744张火煤和扬尘图像构成本实验的数据集，考虑到灯光和雾等一些干扰因素的C1SSS/2C1S/2S/2S/2Concat4C1ConvolutionS/2S/2C2图1SPD ConvFig.1SPD Conv图2BottleNeckCSP模块结构Fig.2BottleNeckCSP module structureConvBottleNeckConv2dConv2dConcatBNLeaky RelUConv禹晨，等：基于YOLOv8化工企业储煤场火煤和扬尘检测方法研究（2）51盐科学与化工第53卷第3期2024年3月影响，数据集中也包含了一些类火和类烟的图像。这些图像具有代表性和多样性，确保模型能够学习到足够多的信息。在收集到图像后，对这些图像进行清洗和预处理，以提高模型的训练效率和效果。随后使用Labelimg工具对图像中的目标物体进行分类、框选等操作标注，并对标注好的图像进行审核，保证标注质量。最后将标注好的数据集划分为训练集、验证集和测试集，以便对模型进行训练、评估和优化，比例为8 1 1，5 420张用作训练集，677张用作测试集和677张用作验证集。3.2实验环境实验环境硬件包括：CPU为Intel（R）Core（TM）i5-13600KF，工作频率为3.5 GHZ，内存大小为32 G，显卡为Nvidia GeForce RTX4060ti16G。软件环境为Python3.8.18，PyTorch 2.0.0，GPU加速器为cuda11.8。预 训 练 模 型 为 YOLOv8s.pt，输 入 图 片 设 置 为640 640，初始学习率设置为0.01，Batch-size设置为4，每次实验设置200个epoch。3.3评价指标研究通过实验评估改进的 Smoke-YOLOv8s算法，将精确率 P（precision）、召回率 R（recall）、平均精 度 AP（average precision）、平 均 精 度 均 值 mAP（mean Average Precision）作为算法的评价指标。公式如（3）、（4）、（5）、（6）所示。P=TPTP+FP（3）R=TPTP+FN（4）AP=01P（R）dR（5）mAP=APdirty+APdamage2（6）式中：AP 为检测的平均精度即为精确率召回率（PrecisionRecall，P-R）曲线的曲线下面积；TP为分类正确的正样本个数；FP为分类错误的正样本个数；FN为分类错误的负样本个数。3.4结果分析3.4.1性能对比实验文章将自建的数据集分别用于原网络的 YOLOv8s 与改进网络的 Smoke-YOLOv8s 模型中进行训练，在同时训练 200 个 epoch 的情况下，得到mAP0.5 变化曲线。由图 5 可知 Smoke-YOLOv8s模型的平均分类性能更好，且在训练 150 个 epoch后结果趋于稳定。图4Smoke-YOLOv8s整体网络结构Fig.4Smoke-YOLOv8s overall network structureBackboneSPDCnovSPDCnovSPDCnovBottleCSPBottleCSPSPDCnovBottleCSPSPDCnovBottleCSPSPPFConcat上采样BottleCSPConcat上采样BottleCSPConcat上采样NeckBottleCSPSPDCnovConcatBottleCSPSPDConvConcatBottleCSPSPDConvConcatBottleCSP160160Dect808040402020图5模型改进前和改进后的 mAP0.5对比情况Fig.5Comparison of mAP 0.5 before and after model improvementmAP0.51.00.80.60.40.20.00255075100 125 150 175 200epoch改进前YOLOv8s改进后Smoke-YOLOv8s图3损失函数迭代过程Fig.3Loss function iterative process76543210-1-2-2-1012345-2-101234576543210-1-2-2-101234576543210-1-276543210-1-2-2-101234576543210-1-2-2-101234576543210-1-2-2-1012345（a）CIoU损失函数迭代过程（b）EIoU损失函数迭代过程523.4.2算法对比实验为进一步验证改进算法与以往算法在烟尘检测上的性能差异，实验选择目标检测领域经典的YOLOv3-tiny、YOLOv5s、YOLOv8s 和Smoke-YOLOv8s算法进行对比实验，所有实验均在同一实验环境、同一数据集下进行，经各算法训练后的模型性能见表 1。由表 1 可知，相比以往的 YOLO 模型，改进的Smoke-YOLOv8s模型在精确率和mAP0.5均有显著的提高，在模型的大小方面，文章改进模型的大小为 25.4 MB，可以很好地满足嵌入式设备对火煤和扬尘检测模型轻量化的要求。3.4.3消融实验为了验证不同的改进策略对原始网络的影响，针对 YOLOv8s 模型进行了 3 组消融实验，即增加SPD Conv 模 块，将 特 征 提 取 模 块 改 为 BottleNeckCSP，将损失函数由 CIoU 改进为 EIoU。实验结果见表 2。在每组实验中，所有实验均在同一个数据集和相同的训练参数下完成。由表 2 可知，在主干特征提取网络Backbone层部分，引入SPD Conv代替原有的普通卷积后，原模型的各项性能均得到明显改善，精确率在原模型基础上提高了1.3%，mAP 0.5 直接在原模型的基础上增长了 2.1%，在特征融合 Neck 层采用 BottleNeckCSP模块替换之前的C2f模块和改进损失函数为 EIoU 后，mAP 0.5 继续增长了 1.8%，达到了97.3%，模型大小降低了1.4 MB，将该模型部署到嵌入式设备中进行实时火煤和扬尘检测，模型的大小、平均分类性能等参考指标都达到较高水平。3.4.4测试结果YOLOv8s 与文章改进算法 Smoke-YOLOv8s 实际运行效果见图6，在火煤和扬尘检测的准确度上有显著提高，而且提高了对火煤和扬尘的低分辨率图像和呈现小目标状态下的检测性能15。4结论文章设计了一种在复杂背景下进行火煤和扬尘检测的改进 YOLOv8s 算法，与 YOLOv8s 算法相比，改进之后的模型平均精度均值达到97.3%，提升了1.2%的精确率，同时降低了模型权重文件的内存存储和对硬件环境的依赖，可以满足储煤场场景下对模型检测精度、轻量化要求。参考文献1REN S Q，HE K M，GIRSHICK R，et al.Faster R-CNN：towardsreal-time object detection with region proposal networksJ.IEEETransactions on Pattern Analysis and Machine Intelligence，2017，39（6）：1 137-1 149.2HE K M，GKIOXARI G，DOLLR P，et al.Mask R-CNN C/Proceedings of IEEE International Conference on Computer Vision.Washington D.C.，USA：IEEE Press，2017：2 980-2 988.3REDMON J，DIVVALA S，GIRSHICK R，et al.You only look once：unified，real-time object detectionC/Proceedings of IEEE Conference on Computer Vision and Pattern Recognition.WashingtonD.C.，USA：IEEE Press，2016：779-788.表1不同模型性能对比Tab.1Performance comparison of different models模型YOLOv3-tinyYOLOv5sYOLOv8sSmoke-YOLOv8s精确率/%91.393.894.996.1F1分数/%83909295mAP0.5/%84.993.293.497.3模型大小/MB24.418.522.525.4表2消融实验结果Tab.2Ablation experiment results模型YOLOv8s改进点1改进点2改进点3SPDConvBottleNeckEIoU精确率/%94.996.296.296.1mAP0.5/%93.495.595.697.3模型大小/MB22.526.825.425.4图6YOLOv8s和Smoke-YOLOv8s检测效果对比Fig.6Comparison of detection performance between YOLOv8sand Smoke-YOLOv8s（a）YOLOv8s检测效果（b）Smoke-YOLOv8s检测效果Smoke&Dust 0.5；t0.5Smoke&Dust 0.7Smoke&Dust 0.3Smoke&Dust 0.3Smoke&Dust 0.5Smoke&Dust 0.8Smoke&Dust 0.8Smoke&Dust 0.8Smoke&Dust 0.8Smoke&Dust 0.9Smoke&Dust 0.9Smoke&Dust 0.9禹晨，等：基于YOLOv8化工企业储煤场火煤和扬尘检测方法研究53盐科学与化工第53卷第3期2024年3月4卢鹏，赵亚琴，陈越，等.复杂背景环境下基于SSD_MobileNet深度学习模型的火焰图像识别研究 J.火灾科学，2020，29（3）：142-149.5HOWARD A，SANDLER M，CHU G，et al.Searching for mobilenetv3C/Proceedings of the IEEE/CVF International Conferenceon Computer Vision，2019：1 314-1 324.6LIU W，ANGUELOV D，ERHAN D，et al.SSD：single shot multiboxdetectorC/European Conference on Computer Vision.Cham：Springer，2016：21-37.7王斌，李靖，赵康，等.面向火焰快速检测的轻量化深度网络研究J.计算机工程与应用，2022，58（17）：256-262.8Bochkovskiy A，Wang C Y，Liao H Y M.YOLOv4：OptimalSpeed and Accuracy of Object Detection J.arXiv，2004，10 934.9H N S，H N L，H N P，et al.Detection and localization of mask occluded faces by transfer learning using faster RCNNJ.SSRNElectronic Journal，2021，45（6）：3 320-3 335.10REDMON J，FARHADI A.YOLO9000：better，faster，strongerC/Proceedings of IEEE Conference on Computer Vision andPattern Recognition.Washington D.C.，USA：IEEE Press，2017：6 517-6 525.11LI C Y，LI L L，JIANG H L，et al.YOLOv6：a single-stage object detection framework for industrial applications Z/OL.（2022-09-07）2023-04-05.12TIAN Z，SHEN C H，CHEN H，et al.FCOS：fully convolutionalone-stage object detection C/Proceedings of the IEEE/CVF International Conference on C omputer Vision.Seoul，Korea：IEEE，2019：9 627-9 636.13刘雄彪，杨贤昭，陈洋，等.基于CIoU改进边界框损失函数的目标检测方法 J.液晶与显示，2023，38（5）：656-665.14杨永鹏，郑艺欣，杨真真，等.一种基于改进YOLOv4的图像目标检测方法 J.科学技术与工程，2023，（2）：4 132-4 658.15鄂毓邦.浅谈煤质检验要点 J.盐科学与化工，2017，46（1）：47-51.（编辑：丁捷）变化及时调整运行工艺参数。7）选择与水质相匹配的阻垢剂，对反渗透稳定运行尤为关键。8）若进水水质或处理水量发生变化，需要及时调整加药量，特别是絮凝剂与阻垢剂的加药量需要严格控制。9）根据现场的实际情况，加药泵的有效加药量需要定期校准，防止无效加药的情况发生。10）当膜系统发生污堵时，若多次采用常规化学清洗无法达到理想的效果时，建议选用专用清洗剂，并由专业的清洗团队进行化学清洗，减少不必要的损失。参考文献1 杨军，贝鑫，李林，等.反渗透膜生物粘泥污堵的化学清洗.中国盐业 J，2022（10）：51-53，56.2 毛维东，周如禄，郭中权.煤矿矿井水零排放处理技术与应用 J.煤炭科学技术，2017，45（11）：205-210.3 赵静，矿井水废水零排放分盐项目的应用 J.化工设计通讯，2022，48（1）：177-179，183.4 胡林.矿井废水处理零排放工艺设计实例J.净水技术，2020，39（7）：41-45.5 李福勤，赵桂峰，朱云浩，等.高矿化度矿井水零排放工艺研究J.煤炭科学技术，2018，46（9）：81-86.6 张洪菊.基于反渗透工艺处理地下水的试验研究J.盐科学与化工，2023，52（9）：22-25.（编辑：李海燕）（上接第48页）中盐工程技术研究院有限公司封一中盐工程技术研究院有限公司封二北京泰维安干燥设备技术开发公司天津中盐海晶投资咨询有限公司期刊合办单位及理事会成员单位封三盐科学与化工 征稿简则封四广告目次54