基于改进YOLOv8的红外图像行人检测算法

An Infrared Image Pedestrian Detection Algorithm Based on an Improved YOLOv8

  • 摘要: 针对当前红外图像行人识别容错率较低、网络模型较大的问题,提出一种基于YOLOv8改进的轻量级红外图像行人检测算法RT_YOLOv8。首先采用RepGhostNet作为主干网络,提升模型在总体上对行人的特征提取能力;引入Triplet注意力机制,通过宽度(W)、高度(H)、通道(C)三维度的交互,使得网络更好地提取目标特征信息;使用可变形卷积DCNv2重新构建C2f模块来提升网络的检测能力;在Detect模块中引入PConv替换部分常规卷积,减少冗余计算和内存访问,有效地提取空间特征。实验结果表明,相较于YOLOv8,RT_YOLOv8行人检测算法在实验建立的红外数据集上mAP50提升1.51%,GFLOPS降低50.62%,满足低算力兼顾高精度的部署要求。

     

    Abstract: A lightweight infrared image pedestrian detection algorithm (RT_YOLOv8) based on YOLOv8 was developed to solve the problems of the low fault tolerance rate and large network model of infrared image pedestrian recognition. First, RepGhostNet was used as the backbone network to improve the overall feature extraction ability of the model for pedestrians. The triplet attention mechanism was introduced to improve the ability of the network to extract the target feature information through the three-dimensional interaction of the width (W), height (H), and channel (C). The deformable convolutional network version 2 (DCNv2) was used to reconstruct the C2f module and improve the detection ability of the network. PConv was added to the detection module to replace a portion of the conventional convolution, and reduce redundant computing and memory access, and the spatial features were effectively extracted. The results of an experiment showed that, compared with YOLOv8, the RT_YOLOv8 pedestrian detection algorithm increased mAP50 by 1.51% and decreased GFLOPS by 50.62% on the infrared dataset established in the experiment, which met the deployment requirements of low computing power and high precision.

     

/

返回文章
返回