长春理工大学学报(自然科学版)2024,Vol.47Issue(2) :75-83.

基于改进YOLOv7-Tiny的交通多目标检测方法

Traffic Multi-target Detection Method Based on Improved YOLOv7-Tiny

许文娟 李野 江晟 王博文
长春理工大学学报(自然科学版)2024,Vol.47Issue(2) :75-83.

基于改进YOLOv7-Tiny的交通多目标检测方法

Traffic Multi-target Detection Method Based on Improved YOLOv7-Tiny

许文娟 1李野 1江晟 1王博文1
扫码查看

作者信息

  • 1. 长春理工大学 物理学院,长春 130022
  • 折叠

摘要

在复杂的多目标交通环境中存在检测种类多、背景信息繁杂、图像分辨率低不能有效检测等问题,使用常见的目标检测算法不能达到高精度的实时检测效果,因此提出一种改进YOLOv7-Tiny的交通多目标检测算法.改进算法中首先使用部分卷积——PConv替换原始卷积,优化模型参数量和运行速度;其次采用轻量级算子CARAFE替换原有上采样部分的最临近插值,提升特征融合能力;最后采用EfficiCLoss替换原有损失函数,提高边界框的定位精度改善检测目标因遮挡而漏检问题.此外创建一个基于交通复杂场景的多目标数据集,在此数据集上进行实验,结果表明改进后的检测算法相较于原YOLOv7-Tiny网络的mAP提高了4.3%,检测速度提高了12.5%,参数量减少了30%,满足智慧交通实时检测的要求.

Abstract

In the complex multi-object traffic environment ,there are challenges such as diverse detection categories ,intricate background information ,and ineffective detection due to low image resolution. Commonly used object detection algorithms fail to achieve high-precision real-time detection. To address these issues ,we propose an improved algorithm for multi-object traffic detection based on YOLOv7-Tiny.In the enhanced algorithm ,we first employ Partial Convolution (PConv) to replace the original convolution ,optimize model parameters and runtime speed. Next ,we integrate the lightweight operator , Context-Aware Reassembly Feature (CARAFE),to replace the previous nearest-neighbor interpolation in the upsampling section ,enhancing feature fusion capabilities. Lastly ,Efficient Classification Loss (EfficiCLoss) is introduced to replace the original loss function ,improving the localization accuracy of bounding boxes and mitigating the issue of missed detections caused by occlusion. Additionally,we create a multi-object dataset based on complex traffic scenarios for experimentation. The results demonstrate that the enhanced detection algorithm achieves a 4.3% improvement in mean Average Precision (mAP) compared to the original YOLOv7-Tiny network. Furthermore ,the detection speed increases by 12.5%,and the parameter count decreases by 30%,meeting the requirements for real-time detection in intelligent transportation systems.

关键词

交通目标检测/YOLOv7-Tiny/Faster-Net/EfficiCLoss

Key words

traffic target detection/YOLOv7-Tiny/Faster-Net/EfficiCLoss

引用本文复制引用

基金项目

国家自然科学基金(12274041)

国家自然科学基金(U2031113)

出版年

2024
长春理工大学学报(自然科学版)
长春理工大学

长春理工大学学报(自然科学版)

CSTPCD
影响因子:0.432
ISSN:1672-9870
参考文献量16
段落导航相关论文