双分支GAN与注意力机制的火灾隐患检测算法.pdfVIP

2282024，60（14）ComputerEngineeringandApplications计算机工程与应用

双分支GAN与注意力机制的火灾隐患检测算法

1，211

李牧，何金诚，杨恒

1.西安理工大学自动化与信息工程学院，西安710048

2.陕西省智能协同网络军民共建重点实验室，西安710048

摘要：针对传统火灾报警在夜间等极端天气下效果不佳，受限于复杂环境等问题，提出一种基于红外与可见光图

像融合的火灾预警算法。在生成对抗网络(GAN)中设计并提出双分支注意力结构。其中一条分支通过密集残差子

网提取更多鲁棒的特征信息，另一条分支通过注意力子网(efficientcoordinatechannelattentiongroup,ECCAG)弥补

空间信息的缺失，以最大限度获取更多高频细节特征，设计并提出了一种调节损失作为损失函数，通过改进GAN算

法得到融合图像，根据提出的火灾预警算法判断是否存在火灾隐患。实验结果表明：改进GAN算法得到的融合数

据集目标检测的平均准确率为96.19%，相较于单一红外数据集与原始GAN算法数据集的目标检测平均准确率分别

提高了11.09个百分点与6.2个百分点，在公开数据集TNO与LLVIP数据集上测试火灾患检测准确率为97.45%。结

果表明，火灾预警算法可以在未发生火灾时及时预警，针对不同场景都可得到显著的检测效果。

关键词：生成对抗网络；图像融合；早期火灾预警；双分支结构；注意力机制

文献标志码：A中图分类号：TP183；TP391doi：10.3778/j.issn.1002-8331.2304-0288

FireHazardDetectionAlgorithmwithDual-BranchGANandAttentionMechanism

1，211

LIMu,HEJincheng,YANGHeng

1.CollegeofAutomationandInformationEngineering,Xi’anUniversityofTechnology,Xi’an710048,China

2.ShaanxiProvincialKeyLaboratoryofIntelligentCollaborativeNetwork,Xi’an710048,China

Abstract：Aimingattheproblemsoftraditionalfirealarmcannotbewarnedbeforetheoccurrenceoffire,theeffectis

notgoodinextremeweathersuchasnight,anditislimitedbycomplexenvironment,afirealarmalgorithmbasedon

infraredandvisiblelightimagesfusionisproposed.Atwo-branchattentionstructureisdesignedandproposedinagenera-

tiveadversarialnetwork(GAN).Onebranchextractsmorerobustfeatureinformationthroughthedenseresidualnetwork,

andtheotherbranchmakesupforthelackofspatialinformationthroughtheefficientcoordinatechannelattentiongroup

(ECCAG)tomaximizetheacquisitionofmorehigh-frequencydetailfeatures,anddesignsandproposesaregulationloss

asalossfunction,andobtainsthefusionimagebyimprovingt