【技术实现步骤摘要】
一种基于改进YOLOv7的烟火识别方法
[0001]本专利技术涉及一种基于改进YOLOv7的烟火识别方法,属于计算机视觉和深度学习
技术介绍
[0002]在电气室的生产运行过程中,烟火是一种常见的安全隐患,为了确保工作场所的安全和稳定,需要对电气室中的烟火进行及时的识别和监测。传统的烟火监测方法需要人工巡检,存在效率低下和漏检等问题。
[0003]现在,随着深度学习技术的发展,利用深度学习技术对监控中发生火灾的场景进行检测成为了可能,这将有助于我们及时准确地知道火灾发生的位置并发出警报来及时做出相应的处理,从而降低火灾所造成的财产等损失。深度网络模型在视觉领域取得了突出进展,将深度网络模型应用于烟火检测成为研究热点,有学者先后提出了Faster Region Convolution Neural Network(Faster
‑
RCNN)、Single Shot MultiBox Detector(SSD)、You Only Look Once(YOLO)等性能良好的网络模型,并成功应用于各种目标检测;但是现有技术下神经网络在图像识别下的应用,受环境因素影响大,且对小尺寸目标的检测也存在一定的局限性。
技术实现思路
[0004]本专利技术所要解决的技术问题是提供一种基于改进YOLOv7的烟火识别方法,基于创新网络结构设计,适用更多处理环境的同时,能够有效提升目标检测精度。
[0005]本专利技术为了解决上述技术问题采用以下技术方案:本专利技术设计了一种基于改 ...
【技术保护点】
【技术特征摘要】
1.一种基于改进YOLOv7的烟火识别方法,其特征在于:执行步骤A至步骤C,获得目标室内所对应的火情检测模型,进而实时执行步骤i,应用火情检测模型针对实时所获目标室内的拍摄图像进行检测,实现目标室内的实时火情检测;步骤A.收集目标室内环境出现火情的各帧拍摄样本图像,并获得各帧拍摄样本图像中的各火情局部图像真实框,以及获得各帧拍摄样本图像分别所对应其拍摄时间下、目标室内环境关于预设各类环境因素的特征值,由单帧拍摄样本图像、以及其所对应拍摄时间下目标室内环境关于预设各类环境因素的特征值,结合该单帧拍摄样本图像中各火情局部图像真实框,构建单个火情图像样本,进而获得各个火情图像样本,然后进入步骤B;步骤B.针对YOLOv7网络添加SEAttention注意力机制模块,构建待训练网络,并进入步骤C;步骤C.基于各个火情图像样本,以拍摄样本图像、该拍摄样本图像所对应拍摄时间下目标室内环境关于预设各类环境因素的特征值为输入,该拍摄样本图像中各火情局部图像检测框为输出,结合关于拍摄样本图像中各火情局部图像真实框下的损失函数,针对待训练网络进行训练,获得训练后的网络,即构成目标室内所对应的火情检测模型;步骤i.获得拍摄图像的拍摄时间下、目标室内环境关于预设各类环境因素的特征值,结合该拍摄图像,应用火情检测模型进行检测,获得该拍摄图像中是否存在火情局部图像、各火情局部图像检测框。2.根据权利要求1所述一种基于改进YOLOv7的烟火识别方法,其特征在于:所述步骤C针对待训练网络训练过程中、关于拍摄样本图像中各火情局部图像真实框下的损失函数如下:L
Focal_ElOU
=IOU
γ
L
EIOU
其中,L
Focal_ElOU
表示损失函数的结果,IOU表示火情局部图像检测框A与火情局部图像真实框B之间的交并比,γ表示Focal损失函数中的可调参数,L
EIOU
表示EIOU损失函数的结果,L
IOU
表示IOU损失函数的结果;L
dis
表示距离损失函数的结果,用于惩罚火情局部图像检测框A与火情局部图像真实框B在位置上的偏差;L
asp
表示比例损失函数的结果,用于惩罚火情局部图像检测框A与火情局部图像真实框B在长宽比例上的偏差,ρ表示平滑系数,b和b
gt
分别表示火情局部图像检测框A和火情局部图像真实框B的中心点坐标,C表示拍摄样本图像的长或宽,w、h分别表示火情局部图像检测框A的宽度和高度,wb
gt
、h
gt
分别表示火情局部图像真实框B的宽度和高度,C
w
和C
h
分别表示宽度和高度的比例系数。3.根据权利要求1所述一种基于改进YOLOv7的烟火识别方法,其特征在于:所述步骤B中,待训练网络包括一级网络、二级网络、以及三级网络,其中,一级网络按数据流方向包括依次串联的CBS层、CBS层、CBS层、CBS层、C7_1层、MP
‑
C3层、SE_A层、MP
‑
C3层、SE_A层、MP
‑
C3层、C7_1层、SPPCSPC层、CBS层、上采样层,该串联结构中顺序第一个CBS层的输入端构成待训练网络的输入端,该串联结构中最后上采样层的输出端构成一级网络的第一输出端,一级网络还包括两个CBS层,一级网络中串联结构顺序第二个SE_A层的输出端串联其中一个CBS层后、构成一级网络的第二输出端,一级网络中串联结构顺序第一个SE_A层的输出端串
联另一个CBS层后、构成一级网络的第三输出端,一级网络中串联结构中SPPCSPC层的输出端构成一级网络的第四输出端;二级网络包括四个第一组合、两个第二组合、以及一个CBS层与一个上采样层,四个第一组合的结构彼此相同,各第一组合分别自其输入端至其输出端串联Concat层与C7_2层,两个第二组合的结构彼此相同,各第二组合分别均包括两路结构,其中一路结构自其输入端至其输出端串联MP层与CBS层,另一路结构自其输入端至其输出端串联两个CBS层,该两路结构的输入端构成第二组合的两输入端,该两路结构的输出端构成第二组合的两输出端,一级网络的第一输出端、第二输出端对接第一个第一组合的输入端,第一个第一组合的输出端依次串联一个CBS层、一个上采样层,且该上采样层输出端对接第二个第一组合的输入端,第二个第一组合的输入端同时对接一级网络的第三输出端,第二个第一组合的输出端构成二级网络的第一输出端,同时第二个第一组合的输出端对接其中一个第二组合的两输入端,该第二组合的两输出端对接第三个第一组合的输入端,第三个第一组合的输入端同时对接第一个第一组合的输出端,第三个第一组合的输出端构成二级网络的第二输出端,同时第三个第一组合的输出端对接另一个第二组合的两输入端,该第二组合的两输出端对接第四个第一组合的输入端,第四个第一组合的输入端同时对接一级网络的第四输出端,第四个第一组合的输出端构成二级网络的第三输出端;三级网络包括三个结构彼此相同的第三组合,各...
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。