融合通道剪枝的轻量化YOLOv8交通标志检测算法
DOI:
CSTR:
作者:
作者单位:

1.青海师范大学计算机学院 西宁 810008; 2.藏语智能全国重点实验室 西宁 810008; 3.辽宁工程技术大学软件学院 葫芦岛 125105;4.辽宁理工学院信息工程学院 锦州 121000

作者简介:

通讯作者:

中图分类号:

TP391.4; TN919.8

基金项目:

国家自然科学基金面上项目(51874166)、国家自然基金青年基金(51904144)、青海师范大学中青年科研基金(2023QZR002)、青海师范大学2025年大学生创新训练计划项目(qhnucxcy2025022, qhnucxcy2025026)资助


Lightweight YOLOv8 traffic sign detection algorithm incorporating channel pruning
Author:
Affiliation:

1.College of Computer, Qinghai Normal University,Xining 810008, China;2.The State Key Laboratory of Tibetan Intelligence,Xining 810008, China; 3.School of Software, Liaoning Technical University,Huludao 125105, China; 4.College of Information Engineering, Liaoning Institute of Science and Engineering,Jinzhou 121000, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对现有交通标志检测算法在边缘计算设备上部署时面临的计算开销大、对小目标和形变目标检测能力不足的问题,本文提出一种融合通道剪枝的轻量化YOLOv8交通标志检测算法LTS-YOLO。首先,在YOLOv8n的基础上,设计了一种结合噪声抑制与深层语义增强的特征融合模块(NSSE),以优化多尺度特征表达并抑制背景干扰。其次,在主干网络中引入多尺度通道注意力(MSCA),并在检测头前集成局部形变注意力(LDA),以分别增强模型对不同尺度目标的感知能力和对几何形变的鲁棒性,从而构建高精度模型TS-YOLO。最后,为实现模型轻量化,基于BatchNorm缩放因子设计通道剪枝策略对TS-YOLO进行压缩,得到最终的LTS-YOLO模型。在TT100K和CCTSDB两个公开数据集上的实验结果表明,相比基线模型YOLOv8n,TS-YOLO在mAP@50.95指标上分别提升了2.5%和1.8%,经过剪枝得到的LTS-YOLO模型,在保持精度优势的同时,参数量与计算量显著降低,验证了本方法的有效性和实用性。

    Abstract:

    To address the challenges of high computational overhead and insufficient detection capability for small and deformed targets when deploying existing traffic sign detection algorithms on edge computing devices, this paper proposes a lightweight YOLOv8-based traffic sign detection algorithm LTSYOLO incorporating channel pruning. First, building upon YOLOv8n, we design a feature fusion module combining noise suppression and deep semantic enhancement (NSSE) to optimize multi-scale feature representation and suppress background interference. Second, we introduce a multi-scale channel attention (MSCA) mechanism into the backbone network and integrate local deformation attention (LDA) before the detection head, enhancing the model′s perception of multi-scale targets and robustness to geometric deformations, thereby resulting in a high-precision model, TS-YOLO. Finally, to achieve model compression, we apply a BatchNorm scaling factor-based channel pruning strategy to compress TS-YOLO, obtaining the final LTS-YOLO model. Experimental results on the TT100K and CCTSDB datasets demonstrate that, compared to the baseline YOLOv8n, TS-YOLO improves mAP@50 by 2.5% and 1.8% on TT100K and CCTSDB, respectively. After pruning, the resulting LTS-YOLO model maintains its accuracy advantage while significantly reducing both parameter count and computational complexity, demonstrating the effectiveness and practicality of the proposed method.

    参考文献
    相似文献
    引证文献
引用本文

艾强,冯永安,王灵超,董丽欣.融合通道剪枝的轻量化YOLOv8交通标志检测算法[J].电子测量技术,2026,49(5):239-250

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-05-08
  • 出版日期:
文章二维码

重要通知公告

①《电子测量技术》期刊收款账户变更公告