基于YOLOv5s的航拍小目標(biāo)檢測(cè)改進(jìn)算法研究
無(wú)線(xiàn)電工程
頁(yè)數(shù): 9 2023-04-11
摘要: 針對(duì)無(wú)人機(jī)航拍時(shí)拍攝的對(duì)象大小不一、種類(lèi)繁雜且容易被建筑遮擋等問(wèn)題,提出了一種基于YOLOv5s的無(wú)人機(jī)目標(biāo)檢測(cè)改進(jìn)算法VA-YOLO。在已有的主干網(wǎng)絡(luò)中添加CA注意力機(jī)制模塊,擴(kuò)大檢測(cè)區(qū)域,獲得更準(zhǔn)確的位置信息;針對(duì)檢測(cè)小目標(biāo)時(shí)尺度不一導(dǎo)致語(yǔ)義丟失的問(wèn)題,添加小目標(biāo)檢測(cè)層與BiFPN結(jié)構(gòu),加深淺層語(yǔ)義與深層語(yǔ)義結(jié)合,以此豐富對(duì)檢測(cè)目標(biāo)的語(yǔ)義信息;使用損失函數(shù)Varifocal...