信息技术与信息化2024,Issue(8) :40-43.DOI:10.3969/j.issn.1672-9528.2024.08.009

基于YOLOv7-tiny的轻量化火焰检测算法

朱康跃
信息技术与信息化2024,Issue(8) :40-43.DOI:10.3969/j.issn.1672-9528.2024.08.009

基于YOLOv7-tiny的轻量化火焰检测算法

朱康跃1
扫码查看

作者信息

  • 1. 三峡大学计算机与信息学院 湖北宜昌 443002
  • 折叠

摘要

针对目前火焰检测算法具有参数量大、检测效果差等问题,提出一种基于YOLOv7-tiny的火焰检测算法.首先,为减少模型的参数量和计算量,在Backbone层和Head层分别融入PConv和GSConv.然后,为降低轻量化网络造成的特征损失,引入CBAM注意力机制.实验结果表明,改进后的参数量和计算量分别下降47.5%和45%,而mAP提升2.1%.结果表明,改进后算法在参数量、计算量、检测精度和检测速度等方面具有明显优势.

关键词

火焰检测/YOLOv7-tiny/轻量化/注意力机制

引用本文复制引用

出版年

2024
信息技术与信息化
山东电子学会

信息技术与信息化

影响因子:0.29
ISSN:1672-9528
段落导航相关论文