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