一種基于小波變換的YOLOv5火災(zāi)檢測(cè)改進(jìn)算法
無線電工程
頁(yè)數(shù): 8 2023-04-20
摘要: 由于復(fù)雜環(huán)境下類煙火物體的干擾,常導(dǎo)致火災(zāi)檢測(cè)誤判。為了提高圖像中火災(zāi)信號(hào)的檢測(cè)精度,減少火災(zāi)誤報(bào),利用傳統(tǒng)光譜分析在火災(zāi)圖像檢測(cè)技術(shù)中的優(yōu)勢(shì),提出了一種基于小波變換的YOLOv5火災(zāi)檢測(cè)改進(jìn)算法。該算法利用二維Haar小波變換提取圖像的光譜特征,將其輸入到Y(jié)OLOv5s的主干網(wǎng)絡(luò)CSPDarknet中,與卷積層進(jìn)行通道上的特征融合,增強(qiáng)煙火的紋理細(xì)節(jié)特征;通過嵌入CA注意力機(jī)...