基于YOLOv5的輕量級霧天目標(biāo)檢測方法
計算機(jī)工程與應(yīng)用
頁數(shù): 11 2023-12-01
摘要: 針對霧天場景下目標(biāo)檢測算法精度較低、模型復(fù)雜度較高,提出一種基于YOLOv5的輕量級霧天目標(biāo)檢測方法。采用感受野注意力模塊(RFAblock)通過交互感受野特征信息,對感受野添加注意力機(jī)制,提高特征提取能力;采用輕量化網(wǎng)絡(luò)Slimneck作為頸部結(jié)構(gòu),在保持精度的同時降低模型參數(shù)和復(fù)雜度;在損失函數(shù)中引入真實框與預(yù)測框之間的角度向量,提高訓(xùn)練速度和推理的準(zhǔn)確性;采用PNMS(p...