|国家科技期刊平台
首页|期刊导航|无线电工程|基于YOLOv8s的城市背景烟火检测算法

基于YOLOv8s的城市背景烟火检测算法OA

中文摘要

针对目前城市背景烟火检测方法存在检测精度不高,易出现误检、漏检和耗时长等问题,提出一种基于YOLOv8s改进的烟火检测算法。引入加权双向特征金字塔(Bi-directional Feature Pyramid Network, BiFPN)增强特征融合,在BiFPN的基础上融合P2特征层提升小目标检测能力,同时添加基于跨空间学习的高效多尺度注意力(Efficient Multi-scale Attention, EMA)模块,突出目标特征同时抑制背景环境的干扰;为了有效利用特征图的语义信息,引入轻量级通用上采样算子——Content-Aware ReAssembly of Features(CARAFE);基于多尺度卷积注意力(Multi-scale Convolutional Attention, MCA)模块设计了一个轻量化的检测头并提升了检测精度;引入分组卷积空间金字塔池化SPPFCSPC_Group模块,在扩大感受野的同时具有更好的特征提取能力。实验结果表明,改进的YOLOv8s算法在基准模型的基础上计算量减少了25%、参数量减少了37.6%、模型权重大小减少了33.2%,平均精度均值(mean Average Precision, mAP)提升了3.4%,基本满足烟火检测的需求。

于泳波;袁栋梁;孙振;朱灵茜;严增兴;鞠瑞文;李庆党;

青岛科技大学信息科学技术学院,山东青岛266061

计算机与自动化

烟火检测双向特征金字塔YOLOv8s多尺度卷积注意力

《无线电工程》 2024 (011)

P.2566-2575 / 10

山东省泰山学者项目(tshw201502042);山东省重大创新工程(2017CXGC0607)。

10.3969/j.issn.1003-3106.2024.11.007

评论