最新資訊

YOLO-PDS:基于改進(jìn)的YOLOv11的無(wú)人機(jī)小目標(biāo)檢測(cè)算法

目標(biāo)檢測(cè)在遙感領(lǐng)域中具有廣泛的應(yīng)用前景。盡管目標(biāo)檢測(cè)算法在自然圖像中取得了明顯的進(jìn)展,但這些方法直接應(yīng)用于遙感圖像時(shí)仍然面臨諸多挑戰(zhàn)。遙感圖像的背景往往比較復(fù)雜且物體較小,導(dǎo)致前景與背景信息的分布極為不平衡。針對(duì)無(wú)人機(jī)圖像小目標(biāo)和物體遮擋的問(wèn)題,提出了一種基于風(fēng)車(chē)狀卷積(PinwheelConv)改進(jìn)的無(wú)人機(jī)小目標(biāo)檢測(cè)算法。為了改進(jìn)模型對(duì)小目標(biāo)的檢測(cè)效果,在骨干網(wǎng)絡(luò)中使用風(fēng)車(chē)狀卷積替換普通卷積來(lái)更好地適應(yīng)小目標(biāo)提取特征,同時(shí)基于風(fēng)車(chē)狀卷積的思想設(shè)計(jì)了C2f-PC模塊來(lái)替換骨干中的C3k2模塊。為解決無(wú)人機(jī)圖像中目標(biāo)遮擋嚴(yán)重的問(wèn)題,創(chuàng)新性地提出了C2f-PDWR模塊來(lái)替換頸部網(wǎng)絡(luò)中的C3k2模塊,來(lái)增強(qiáng)模型的特征融合能力,同時(shí)引入了SEAM(Spatially Enhanced Attention Module)來(lái)改善模型對(duì)被遮擋物體的檢測(cè)效果。最后,基于YOLOv11提出對(duì)小目標(biāo)檢測(cè)更加高效的YOLO-PDS模型。其在VisDrone2019數(shù)據(jù)集上所提方法較基準(zhǔn)模型YOLOv11檢測(cè)方法mAP50提高3.7%以上,召回率提高2.2%以上。

發(fā)表于:12/17/2025