基于YOLOv7-tiny改进的交通标志小目标实时检测算法

牟家宇 ,  南新元

科学技术与工程 ›› 2024, Vol. 24 ›› Issue (30) :13072-13079

PDF (9877KB)
科学技术与工程 ›› 2024, Vol. 24 ›› Issue (30) :13072-13079 DOI: 10.12404/j.issn.1671-1815.2400306
自动化技术、计算机技术

基于YOLOv7-tiny改进的交通标志小目标实时检测算法

作者信息 +

Improved Real-time Traffic Sign Small Target Detection Algorithm Based on YOLOv7-tiny

Author information +
文章历史 +

摘要

在自然环境下精确实时地检测交通标志小目标对自动驾驶和智慧交通有着重要意义,然而现有算法难以平衡速度与精度的问题。基于YOLOv7-tiny算法,提出了一种改进YOLOv7-tiny的交通标志小目标实时检测算法,即YOLO-T算法。采用条件参数化卷积(CondConv)结构,提升了骨干网络的特征提取能力。为增强对小目标的定位准确度并保证检测速度,设计了TinyFPN特征融合网络结构和ELAN-P网络聚合层。为了验证YOLO-T算法的有效性,在TT100K数据集上做了消融实验和对比实验。实验结果表明,在训练样本及训练设备参数相同的情况下,YOLO-T比YOLOv7-tiny算法的均值平均精度(mAP)提升了16.8%,并且单张图片的检测时间仅10.2 ms。可见,所提的YOLO-T算法能够平衡交通标志小目标的检测速度与精度。

Abstract

Detecting traffic sign small targets accurately and in real time in natural environment is of great significance for automatic driving and intelligent transportation, however, the existing algorithms are difficult to balance the speed and accuracy. Based on the YOLOv7-tiny algorithm, a real-time traffic sign small target detection algorithm that improves YOLOv7-tiny was proposed, namely YOLO-T algorithm. CondConv (conditional parameterized convolution) structure was used to enhance the feature extraction capability of the backbone network. To enhance the localization accuracy of small targets and ensure the detection speed, the TinyFPN feature fusion network structure and ELAN-P network aggregation layer were designed. To verify the effectiveness of the YOLO-T algorithm, ablation experiments and comparison experiments were done on the TT100K dataset. The experimental results show that YOLO-T improves mAP (mean average precision) by 16.8% over the YOLOv7-tiny algorithm with the same training samples and training device parameters, and the detection time of a single image is only 10.2 ms. Therefore, the proposed YOLO-T algorithm is able to balance the speed and accuracy of the detection of the small targets of traffic signs.

关键词

交通标志检测 / 小目标 / YOLO-T算法 / YOLOv7-tiny算法

Key words

traffic sign detection / YOLO-T network / deep learning / YOLOv7-tiny network

引用本文

引用格式 ▾
牟家宇, 南新元. 基于YOLOv7-tiny改进的交通标志小目标实时检测算法[J]. 科学技术与工程, 2024, 24(30): 13072-13079 DOI:10.12404/j.issn.1671-1815.2400306
MU Jia-yu, NAN Xin-yuan. Improved Real-time Traffic Sign Small Target Detection Algorithm Based on YOLOv7-tiny[J]. Science Technology and Engineering, 2024, 24(30): 13072-13079 DOI:10.12404/j.issn.1671-1815.2400306

登录浏览全文

4963

注册一个新账户 忘记密码

参考文献

基金资助

国家自然科学基金(52065064)

PDF (9877KB)

35

访问

0

被引

详细

导航
相关文章

/

〈 〉