基于改进YOLOv7的火灾目标检测算法

龚文强, 肖铎, 何斌, 葛瑞泽

燃烧科学与技术 ›› 2024, Vol. 30 ›› Issue (04) : 394-402.

基于改进YOLOv7的火灾目标检测算法

  • 龚文强, 肖铎, 何斌, 葛瑞泽
作者信息 +
History +

摘要

提出基于改进YOLOv7的火灾目标检测算法,以目标检测网络YOLOv7为基本网络,采用改进后的轻量化网络EfficientNetV2-S替换其主干网络,从而降低模型的参数量.引入坐标注意力机制,对EfficientNetV2-S的卷积模块进行改进.对损失函数进行改进,提升火灾识别的精确率.标注火灾目标检测数据集共6 150张对模型进行训练,与常见火灾目标检测算法对比验证,并进行消融实验.结果表明,该算法的检测精度和识别速度比传统目标检测算法有优势,识别精确率比原YOLOv7算法提升6.5%,识别速度提升28.6%.

关键词

火灾检测 / 目标检测 / 注意力机制 / YOLOv7 / EfficientNetV2

中图分类号

X932 / TP183 / TP391.41

引用本文

导出引用
龚文强, 肖铎, 何斌, 葛瑞泽. 基于改进YOLOv7的火灾目标检测算法. 燃烧科学与技术. 2024, 30(04): 394-402

基金

浙江省重点研发计划资助项目(2019C01150)

评论

Accesses

Citation

Detail

段落导航
相关文章

/