基于AS-YOLOv7的红外目标检测算法

王艳梅, 李明明, 黄月, 张艳珠

PDF(1005 KB)
PDF(1005 KB)
沈阳理工大学学报 ›› 2024, Vol. 43 ›› Issue (04) : 1-6+12.

基于AS-YOLOv7的红外目标检测算法

  • 王艳梅, 李明明, 黄月, 张艳珠
作者信息 +
History +

摘要

为解决传统目标检测算法在红外场景下的目标识别精度低、检测速度差等问题,提出基于AS-YOLOv7的红外目标检测方法。首先针对红外图像的特性,对YOLOv7模型中的特征提取网络进行重构并构建卷积块CBH,保证模型的稳定性和计算速度;然后改进金字塔池化(SPPCSPC)模块,使模型在不增加任何参数的同时提高模型性能;最后将原模型中的损失函数替换为具有动态非单调的WIoU损失,解决红外数据集中质量较好和质量较差的样本间的不平衡问题,提高模型的泛化性。本文实验在自行采集的数据集上进行训练和验证,与原模型相比,AS-YOLOv7的平均精度提升了0.2%,参数量降低了79%,计算量降低了85%,推理时间降低了32 ms。改进后的模型在保持准确度的情况下提高了模型的推理速度。

关键词

红外图像 / YOLOv7 / 目标检测 / 特征提取

中图分类号

TP391.41 / TN219

引用本文

导出引用
王艳梅, 李明明, 黄月, 张艳珠. 基于AS-YOLOv7的红外目标检测算法. 沈阳理工大学学报. 2024, 43(04): 1-6+12

基金

辽宁省教育厅高等学校基本科研项目(LJKZ0245); 装备预研国防科技重点实验室基金项目

评论

PDF(1005 KB)

Accesses

Citation

Detail

段落导航
相关文章

/