一種改進(jìn)YOLOv5s的森林火災(zāi)煙霧檢測算法
計(jì)量學(xué)報(bào)
頁數(shù): 10 2024-09-23
摘要: 提出一種基于改進(jìn)YOLOv5s的森林火災(zāi)煙霧檢測算法。構(gòu)建包含16573幅圖片的火焰煙霧數(shù)據(jù)集,解決訓(xùn)練數(shù)據(jù)不足的問題,提高訓(xùn)練模型的泛化能力。設(shè)計(jì)一種輕量化的GC-C3模塊替換原有的C3模塊,減少模型參數(shù)量和計(jì)算量;將加權(quán)雙向特征金字塔網(wǎng)絡(luò)結(jié)構(gòu)融合到Neck結(jié)構(gòu)中,增強(qiáng)網(wǎng)絡(luò)對于中小目標(biāo)的檢測能力;修改網(wǎng)絡(luò)空間金字塔池化結(jié)構(gòu),使用SimSPPF結(jié)構(gòu)替換SPPF,提高了網(wǎng)絡(luò)的計(jì)算... (共10頁)