[1]郭伟,唐思涛*,王春艳.基于YOLOv7道路交通热红外图像目标检测算法[J].计算机技术与发展,2024,34(11):43-50.[doi:10.20165/j.cnki.ISSN1673-629X.2024.0223]
　GUO Wei,TANG Si-tao*,WANG Chun-yan.Object Detection Algorithm of Road Traffic Thermal Infrared Image Based on YOLOv7[J].,2024,34(11):43-50.[doi:10.20165/j.cnki.ISSN1673-629X.2024.0223]
点击复制

基于YOLOv7道路交通热红外图像目标检测算法()

《计算机技术与发展》[ISSN:1006-6977/CN:61-1281/TN]

卷:
34
期数:
2024年11期
页码:
43-50
栏目:
媒体计算
出版日期:
2024-11-10

文章信息/Info

Title:
Object Detection Algorithm of Road Traffic Thermal Infrared Image Based on YOLOv7
文章编号:
1673-629X(2024)11-0043-08
作者:
郭伟; 唐思涛*; 王春艳
辽宁工程技术大学 软件学院,辽宁 葫芦岛 125105
Author(s):
GUO Wei; TANG Si-tao*; WANG Chun-yan
School of Software,Liaoning Technical University,Huludao 125105,China
关键词:
目标检测; 热红外图像; YOLOv7; ELAN-P; 坐标注意力机制; WIoU
Keywords:
object detection; thermal infrared imaging; YOLOv7; ELAN-P; coordinate attention mechanism(CA); WIoU
分类号:
TP391.7
DOI:
10.20165/j.cnki.ISSN1673-629X.2024.0223
摘要:
热红外图像具有分辨率低、高噪声、空间相关性强的特点。 为解决道路交通红外目标检测算法存在的检测精度低、漏检和误检等问题,提出一种改进 YOLOv7 的算法。 将主干网络原有的 ELAN 模块替换成 ELAN-P 模块,降低模型的参数量和计算量,使模型更加轻量化,增强对红外目标的提取能力;在主干网络和颈部网络引入 CA 注意力机制,将坐标信息嵌入到通道中,增强对模糊目标和密集目标的定位能力;将原有的 CIoU 损失函数替换成 WIoU 损失函数,提高对遮挡目标和弱小目标的检测精度。 在中国热红外数据集 CTIR 上实验表明,改进算法相较于 YOLOv7 算法,参数量和计算量分别减少 11. 6 百分点和 19. 5 百分点,检测精度 mAP 值提高了 3. 1 百分点,其中 Car、Pedestrian、Cyclist、Bus 和 Truck 五个类别的检测结果 AP 值分别提高了 1. 9 百分点、1. 9 百分点、1. 5 百分点、4. 9 百分点和 5. 3 百分点,检测性能有所提升。 在公开数据集 FLIR 上进行泛化性对比实验,结果表明改进算法具有通用性。
Abstract:
Thermal infrared images have the characteristics of low resolution, high noise and strong spatial correlation. To solve the problems of low detection accuracy, missed detection and false detection in infrared target detection algorithms for road traffic, an improved YOLOv7 algorithm was proposed. The ELAN module of the backbone network was replaced by the ELAN-P module,which reduced the number of parameters and calculation of the model,made the model more lightweight,and enhanced the ability to extract infrared targets. The CA mechanism was introduced into the backbone network and neck network,and the coordinate information was embedded into the channel to enhance the localization ability of fuzzy targets and dense targets. The original CIoU loss function was replaced with the WIoU loss function to improve the detection accuracy of occluded targets and dim and small targets. Experiments on the Chinese thermal infrared dataset CTIR show that compared with the YOLOv7 algorithm,the improved algorithm reduces the parameter amount and calculation amount by 11. 6 percentage points and 19. 5 percentage points respectively,and improves the detection accuracy mAP value by 3. 1 percentage points. Among them, the AP value of the five categories of Car,Pedestrian,Cyclist,Bus and Truck increased by 1. 9 percentage points,1. 9 percentage points,1. 5 percentage points,4. 9 percentage points and 5. 3 percentage points respec-tively,indic.

相似文献/References:

[1]刘晓明 李毓蕙 高燕 郑华强.基于目标区域清晰显示的H.264编码策略[J].计算机技术与发展,2010,(06):29.
　LIU Xiao-ming,LI Yu-hui,GAO Yan,et al.A Coding Strategy of H.264 Based on High-definition Display of Target Region[J].,2010,(11):29.
[2]刘翔 吴谨 祝愿博 康晓晶.基于视频序列的目标检测与跟踪技术研究[J].计算机技术与发展,2009,(11):179.
　LIU Xiang,WU Jin,ZHU Yuan-bo,et al.A Study of Object Detecting and Tracking Based on Video Sequences[J].,2009,(11):179.
[3]曙光 张超 蔡则苏.基于改进的混合高斯模型的目标检测方法[J].计算机技术与发展,2012,(07):60.
　SHU Guang,ZHANG Chao,CAI Ze-su.Target Detection Method Based on Improved Gaussian Mixture Model[J].,2012,(11):60.
[4]刘洁,李目,周少武.一种混沌混合粒子群优化RBF神经网络算法[J].计算机技术与发展,2013,(08):181.
　LIU Jie[],LI Mu[],ZHOU Shao-wu[].An Algorithm of Chaotic Hybrid Particle Swarm Optimization Based on RBF Neural Network[J].,2013,(11):181.
[5]蒋翠清,孙富亮,吴艿芯. 基于相对欧氏距离的背景差值法视频目标检测[J].计算机技术与发展,2015,25(01):37.
　JIANG Cui-qing,SUN Fu-liang,WU Nai-xin. Video Object Detection of Background Subtraction Method Based on Relative Euclidean Distance[J].,2015,25(11):37.
[6]卢官明,衣美佳. 步态识别关键技术研究[J].计算机技术与发展,2015,25(07):100.
　LU Guan-ming,YI Mei-jia. Research on Critical Techniques in Gait Recognition[J].,2015,25(11):100.
[7]高翔,朱婷婷,刘洋. 多摄像头系统的目标检测与跟踪方法研究[J].计算机技术与发展,2015,25(07):221.
　GAO Xiang,ZHU Ting-ting,LIU Yang. Research of Target Detection and Tracking Method for Multi-camera System[J].,2015,25(11):221.
[8]章文洁[][],黄旻[],张桂峰[]. 滤光片多光谱成像中运动目标场景误配准修正[J].计算机技术与发展,2016,26(01):18.
　ZHANG Wen-jie[][],HUANG Min[],ZHANG Gui-feng[]. Misregistration Correction for Moving Object Scene in Filter-type Multispectral Imaging[J].,2016,26(11):18.
[9]施泽浩,赵启军.基于全卷积网络的目标检测算法[J].计算机技术与发展,2018,28(05):55.[doi:10.3969/j.issn.1673－629X.2018.05.013]
　SHI Ze-hao,ZHAO Qi-jun.Object Detection Algorithm Based on Fully Convolutional Neural Network[J].,2018,28(11):55.[doi:10.3969/j.issn.1673－629X.2018.05.013]
[10]张夏清,茅耀斌. 一种改进的ViBe背景提取算法[J].计算机技术与发展,2016,26(07):36.
　ZHANG Xia-qing,MAO Yao-bin. An Improved ViBe Background Generation Method[J].,2016,26(11):36.

更新日期/Last Update: 2024-11-10