一種改進(jìn)YOLOv5s的森林火災(zāi)煙霧檢測(cè)算法
摘要: 提出一種基于改進(jìn)YOLOv5s的森林火災(zāi)煙霧檢測(cè)算法。構(gòu)建包含16573幅圖片的火焰煙霧數(shù)據(jù)集,解決訓(xùn)練數(shù)據(jù)不足的問(wèn)題,提高訓(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ò)對(duì)于中小目標(biāo)的檢測(cè)能力;修改網(wǎng)絡(luò)空間金字塔池化結(jié)構(gòu),使用SimSPPF結(jié)構(gòu)替換SPPF,提高了網(wǎng)絡(luò)的計(jì)算... (共10頁(yè))
開(kāi)通會(huì)員,享受整站包年服務(wù)