• 《工程索引》(EI)刊源期刊
    • 中文核心期刊
    • 中國科技論文統計源期刊
    • 中國科學引文數據庫來源期刊

    留言板

    尊敬的讀者、作者、審稿人, 關于本刊的投稿、審稿、編輯和出版的任何問題, 您可以本頁添加留言。我們將盡快給您答復。謝謝您的支持!

    姓名
    郵箱
    手機號碼
    標題
    留言內容
    驗證碼

    基于改進YOLOv5的安全帽檢測算法

    侯公羽 陳欽煌 楊振華 張又文 張丹陽 李昊翔

    侯公羽, 陳欽煌, 楊振華, 張又文, 張丹陽, 李昊翔. 基于改進YOLOv5的安全帽檢測算法[J]. 工程科學學報. doi: 10.13374/j.issn2095-9389.2022.12.07.002
    引用本文: 侯公羽, 陳欽煌, 楊振華, 張又文, 張丹陽, 李昊翔. 基于改進YOLOv5的安全帽檢測算法[J]. 工程科學學報. doi: 10.13374/j.issn2095-9389.2022.12.07.002
    Helmet detection method based on improved YOLOv5[J]. Chinese Journal of Engineering. doi: 10.13374/j.issn2095-9389.2022.12.07.002
    Citation: Helmet detection method based on improved YOLOv5[J]. Chinese Journal of Engineering. doi: 10.13374/j.issn2095-9389.2022.12.07.002

    基于改進YOLOv5的安全帽檢測算法

    doi: 10.13374/j.issn2095-9389.2022.12.07.002
    詳細信息
    • 中圖分類號: TP391.4

    Helmet detection method based on improved YOLOv5

    • 摘要: 為了解決建筑工地、隧道、煤礦等施工場景中現有安全帽檢測算法對于小目標、密集目標以及復雜環境下的檢測精度低的問題,提出了一種基于YOLOv5的改進目標檢測算法,記為YOLOv5-GBC。首先使用Ghost卷積對骨干網絡進行重構,使得模型的復雜度有了顯著降低。其次使用雙向特征金字塔網絡(BiFPN)加強特征融合,使得算法對小目標和密集目標準確率提升。最后引入坐標注意力(Coordinate Attention)模塊,能夠將注意力資源分配給關鍵區域,從而在復雜環境中降低背景的干擾。為了驗證算法的可行性,以課題組收集的安全帽數據集為基礎,選用了多種經典算法進行對比,并且進行了消融實驗,探究各個改進模塊的提升效果。實驗結果表明:改進算法YOLOv5-GBC相較于YOLOv5s算法,算法平均精確率(IOU=0.5)提升了4.9%,達到了93.6%,檢測速度達到了124.3FPS,模型更加輕量化,在密集場景和小目標場景下檢測能力提升顯著,并且同時滿足安全帽檢測精度和實時性的要求,給復雜施工環境下安全帽檢測提供了一種新的方法。

       

    • 加載中
    計量
    • 文章訪問數:  201
    • HTML全文瀏覽量:  19
    • PDF下載量:  51
    • 被引次數: 0
    出版歷程
    • 網絡出版日期:  2023-05-05

    目錄

      /

      返回文章
      返回
      中文字幕在线观看