基于概率兩階段CenterNet2的林火圖像預(yù)警檢測(cè)方法
計(jì)算機(jī)工程與科學(xué)
頁(yè)數(shù): 7 2023-10-15
摘要: 森林火災(zāi)的及時(shí)預(yù)警對(duì)于森林保護(hù)有著至關(guān)重要的作用,由于林中煙火背景復(fù)雜、干擾因素較多,檢測(cè)精度和效率均會(huì)受到影響。為此提出了一種基于CenterNet2的林火圖像檢測(cè)方法,采用輕量級(jí)主干網(wǎng)絡(luò)VoVNetV2結(jié)合非對(duì)稱卷積核來(lái)提高特征提取能力和檢測(cè)速度,同時(shí)在加權(quán)雙向特征金字塔網(wǎng)絡(luò)中引入注意力機(jī)制eSE進(jìn)行特征融合,提高小目標(biāo)檢測(cè)精度,最后采用SIoU損失函數(shù)提升目標(biāo)框回歸效果。...