一种基于CA-SRGAN的古建筑消防危险品检测模型  

A CA-SRGAN-based fire-hazardous item detection model in ancient architecture

作　　者：李悦明 黄国忠[1] 王波[1] 高学鸿 孙占辉 潘睿 LI Yueming;HUANG Guozhong;WANG Bo;GAO Xuehong;SUN Zhanhui;PAN Rui(Research Institute of Macro-Safety Science,University of Science and Technology Beijing,Beijing 100083,China;Institute of Public Safety Research,Department of Engineering Physics,Tsinghua University,Beijing 100084,China)

机构地区：[1]北京科技大学大安全科学研究院,北京100083 [2]清华大学工程物理系,公共安全研究院,北京100084

出　　处：《清华大学学报（自然科学版）》2024年第11期1870-1879,共10页Journal of Tsinghua University(Science and Technology)

基　　金：国家重点研发计划项目(2021YFC1523502)。

摘　　要：2010—2022年,古建筑火灾事故频繁发生,对珍贵历史文物的保护提出了巨大挑战。为应对这一挑战,该研究提出一种基于图像超分辨率技术的消防危险品检测模型,用于识别易燃物品和可燃物品。在实际应用中,应准确归类危险品,并精确定位危险品的位置,由于普通摄像头图像的分辨率均较低,因而采用图像超分辨率生成对抗网络(super-resolution generative adversarial network,SRGAN)算法重建图像;此外,引入坐标注意力机制提高检测精度,并使用完整交并比(complete intersection over union,CIoU)损失函数加速模型训练。这一综合方法可有效检测古建筑区域内的消防危险品。使用测试数据集对该模型进行了一系列实验,结果表明:改进后的模型具有0.830的平均精度均值(mean average precision,mAP)和0.940的精确率;在服务器上,该模型的推理速度高达92 fps。该文提出模型有效且高效,为古建筑火灾防控工作提供了强大的技术支持。[Objective] Ancient buildings are not just architectural structures but valuable cultural heritage sites that,when destroyed by fires,cause irreplaceable losses to history.In many instances,the root cause of these fires is not identified,although fire-hazardous items(such as cigarettes) often emerge as possible culprits.At present,fire risk assessment methods often fall short,offering subjective results and low detection rates that do not often meet the needs of practical application requirements.Therefore,we adopt a approach:real-time detection of fire-hazardous items such as cigarettes for effective monitoring and early warning.We aim to establish a model that provides accurate detection in real time using existing monitoring systems and video images.Considerably,we also plan a method for reconstructing blurred images.[Methods] To detect fire-hazardous items,five categories are selected:lighters,matches,candles,cigarettes,and incense.We independently create a dataset in VOC format,including 7 157 and 6 536 images of hazardous and non-hazardous items.Regarding the model,we choose the you only look once v5(YOLOv5) model as the starting point and made two key enhancements.First,we improve the generalized intersection over union(GIoU) loss function used in the YOLOv5 model by replacing it with complete intersection over union(CIoU),a new loss function.Unlike GIoU,CIoU considers the distance of the center point and aspect ratio,providing an accurate assessment of the quality of the predicted bounding box and leading to enhanced target box regression.Second,we tackle the issue of mismatched weight distribution between feature maps and channels by incorporating coordinate attention(CA) into the YOLOv5 backbone.CA reduces the number of channels in the feature map to increase the receptive field,learns the weight distribution of channels,and reallocates channel features based on these parameters.Ultimately,to reconstruct images,we utilize a super-resolution reconstruction algorithm based on the generative adversarial ne

关 键 词：古建筑 消防危险品 CA-SRGAN 深度学习 

分 类 号：TP18[自动化与计算机技术—控制理论与控制工程]