一種改進YOLOv5s的森林火災煙霧檢測算法
計量學報
頁數(shù): 10 2024-09-23
摘要: 提出一種基于改進YOLOv5s的森林火災煙霧檢測算法。構建包含16573幅圖片的火焰煙霧數(shù)據(jù)集,解決訓練數(shù)據(jù)不足的問題,提高訓練模型的泛化能力。設計一種輕量化的GC-C3模塊替換原有的C3模塊,減少模型參數(shù)量和計算量;將加權雙向特征金字塔網(wǎng)絡結構融合到Neck結構中,增強網(wǎng)絡對于中小目標的檢測能力;修改網(wǎng)絡空間金字塔池化結構,使用SimSPPF結構替換SPPF,提高了網(wǎng)絡的計算... (共10頁)
開通會員,享受整站包年服務