首页 | 本学科首页   官方微博 | 高级检索  
     

基于级联稀疏查询机制的轻量化火灾检测算法
引用本文:张小雪,王雨,吴思远,孙帮勇.基于级联稀疏查询机制的轻量化火灾检测算法[J].光电工程,2023(10):107-122.
作者姓名:张小雪  王雨  吴思远  孙帮勇
作者单位:1. 西安理工大学印刷包装与数字媒体学院;2. 中国科学院西安光学精密机械研究所光谱成像技术重点实验室
基金项目:国家自然科学基金资助项目(62076199);;陕西省重点研发计划项目(2022ZDLGY01-03);;瞬态光学与光子技术国家重点实验室开放基金项目(SKLST202214);;陕西省教育厅重点科学研究计划项目(23JY063);;西安市科技计划项目(22GXFW0088)~~;
摘    要:针对现有火灾检测算法仍存在的模型复杂、检测速度慢、误检率高等问题,提出一种基于级联稀疏查询机制的轻量化火灾检测网络LFNet。首先,建立了轻量化的图像特征提取模块ECDNet,其通过在YOLOv5s主干网络中嵌入轻量化注意力模块ECA (efficient channel attention),用于解决火灾检测中火焰与烟雾的多尺度难点;其次,利用深层特征提取模块FPN+PAN,对不同层级的特征图进行深度处理和多尺度融合;最后,利用嵌入轻量化的级联稀疏查询模块CSQ (cascade sparse query)提升对早期火灾中的小火焰与薄烟雾的检测准确率。实验表明,本文方法在mAP和Precision等客观指标上的综合表现达到最优,同时在实现较高检测精度时的参数量也较低,能够满足实际场景的火灾检测要求。

关 键 词:目标检测  火灾检测  轻量化  级联稀疏查询机制  Slimming
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号