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