面向無人機航拍圖像小目標(biāo)檢測方法

打開文本圖片集

摘要：針對航拍圖像目標(biāo)檢測中小目標(biāo)特征模糊問題，提出一種改進(jìn)YOLO_v5x的目標(biāo)檢測算法。通過在YOLO_v5x的主干和頸部網(wǎng)絡(luò)中添加空間到深度（space-to-depth，SPD）模塊來減少細(xì)粒度信息丟失；在檢測輸出端添加1個小目標(biāo)預(yù)測頭，提高算法學(xué)習(xí)低分辨率特征的效率；引入?yún)f(xié)調(diào)注意力（coordinate attention，CA）機制，將橫向和縱向的位置信息編碼到通道注意中，增強網(wǎng)絡(luò)對不同維度特征的提取能力；在完整交并比 （complete-intersection over union，CIOU）損失函數(shù)的基礎(chǔ)上引入Alpha交并比（α?IOU）損失函數(shù)，獲得更準(zhǔn)確的邊界框回歸，實現(xiàn)圖像中目標(biāo)更精確的定位。（剩余12052字）