基于改進YOLOv5的小目標煙霧檢測算法
中國安全科學學報
頁數(shù): 7 2023-11-15
摘要: 為解決火災中的小目標煙霧檢測精度不高的問題,提出一種基于改進YOLOv5的小目標煙霧檢測算法。首先,將特征融合注意力(FFA)模塊引入至主干網(wǎng)絡中,使模型專注于小目標煙霧特征信息的提??;其次,通過采用多尺度金字塔解耦頭(MPDH)模塊替換卷積層模塊,以改進YOLOv5算法中預測頭層的檢測部分,用于提升小目標煙霧的定位精度;最后,在專有數(shù)據(jù)集上進行試驗驗證與分析。結(jié)果表明:基于改... (共7頁)