基于YOLOv8s改進的輕量化交通標志檢測模型DP-YOLO:針對小目標檢測的優化與性能提升

《Digital Signal Processing》:DP-YOLO: A lightweight traffic sign detection model for small object detection

【字體: 時間:2025年05月16日 來源:Digital Signal Processing 2.9

編輯推薦:

  為解決自動駕駛場景中交通標志小目標檢測精度不足和模型復雜度高的問題,研究人員提出基于YOLOv8s改進的DP-YOLO模型。通過移除大目標檢測層、新增小目標檢測層,設計DBBNCSPELAN4特征提取模塊和PTCSP混合架構,結合W3F_MPDIoU損失函數,在TT100K數據集上實現mAP0.5提升5.8%的同時參數減少77.0%,為邊緣端部署提供高效解決方案。

  

在自動駕駛技術快速發展的今天,交通標志識別系統如同車輛的"交通語言翻譯官",其準確性直接關系到行車安全。然而現實場景中,交通標志往往以"迷你尺寸"出現在復雜環境中——不足32×32像素的小標志被樹木遮擋、因車速模糊、受光線干擾,傳統檢測算法就像用漁網撈芝麻,漏檢誤檢頻發。更棘手的是,車載設備的算力限制要求模型必須"瘦身",但現有輕量化方案如YOLOv8n、LeYOLO等在壓縮參數時,又像過度節食的運動員,犧牲了關鍵的特征提取能力。這種"檢測精度"與"模型輕量"的拉鋸戰,成為制約自動駕駛落地的關鍵瓶頸。

針對這一難題,廣西民族大學的研究團隊在《Digital Signal Processing》發表研究,提出DP-YOLO模型。通過解剖YOLOv8s的結構缺陷,研究人員實施了三項精準"外科手術":首先切除冗余的大目標檢測層,移植高分辨率小目標檢測層;隨后設計DBBNCSPELAN4模塊強化特征提取,創新PTCSP架構融合CNN與Transformer優勢;最后開發W3F_MPDIoU損失函數提升收斂效率。就像為模型配備"顯微鏡頭"和"節能芯片",在TT100K、GTSDB和CCTSDB數據集上分別實現mAP0.5提升5.8%、2.7%和1.3%,參數量驟降77%。

關鍵技術包括:1)結構重參數化技術將DBBNCSPELAN4模塊的六種卷積分支合并為單一推理結構;2)PTCSP模塊采用3:1通道分配策略平衡CNN與Transformer計算開銷;3)W3F_MPDIoU融合最小點距(MPDIoU)、困難樣本聚焦(Focaler-IoU)和動態加權(WIoUv3)機制;實驗使用TT100K數據集9404張圖像,按7:2:1劃分訓練/測試/驗證集。

【網絡架構革新】移除P5大目標檢測層并新增P2檢測層,使特征圖分辨率從80×80提升至160×160,小目標檢測APs提升6.4%,模型體積從22.6MB壓縮至5.7MB。如圖3所示,改進后的特征圖能清晰保留5像素級標志輪廓。

【特征提取優化】DBBNCSPELAN4模塊通過1×1卷積與3×3卷積的四種分支組合,在訓練階段增強多尺度特征捕獲,推理時轉化為單一3×3卷積。相比原C2f模塊,參數量減少8.1%的同時,mAP0.5提升0.8%。

【混合架構設計】PTCSP模塊將30%通道分配給Transformer分支(MHSA_CGLU結構),70%保留給CNN分支。這種"分而治之"策略使GFLOPs降低3.5%,其中CGLU門控機制增強局部特征建模,DropPath操作防止過擬合。

【損失函數創新】W3F_MPDIoU通過公式LW3F_MPDIoU=rRWIoULF_MPDIoU實現動態加權,其中β=L?IoU/LIoU構建單調聚焦系數。在TT100K數據集上比CIoU提升1.8% mAP0.5,特別對"禁止通行"等稀有標志識別率提高12.3%。

這項研究開創性地實現了"檢測精度不降反升,模型體積大幅縮減"的雙贏局面。其價值如同為自動駕駛系統打造了專用"顯微鏡"——既能捕捉毫米級標志細節,又滿足車載芯片的嚴苛算力限制。特別是PTCSP模塊中Transformer與CNN的黃金比例分配,為多模態架構設計提供新范式。未來通過TensorRT加速和知識蒸餾技術,有望在Jetson等邊緣設備實現毫秒級推理,推動交通標志識別從實驗室走向真實復雜路況。

相關新聞
生物通微信公眾號
微信
新浪微博
  • 急聘職位
  • 高薪職位

知名企業招聘

熱點排行

    今日動態 | 人才市場 | 新技術專欄 | 中國科學人 | 云展臺 | BioHot | 云講堂直播 | 會展中心 | 特價專欄 | 技術快訊 | 免費試用

    版權所有 生物通

    Copyright© eBiotrade.com, All Rights Reserved

    聯系信箱:

    粵ICP備09063491號

    亚洲欧美自拍偷拍,亚洲人成77777,亚洲男女自偷自拍,亚洲成年在线