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