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