一種基于多頭自注意力機(jī)制的無人機(jī)變尺寸目標(biāo)檢測(cè)方法,屬于目標(biāo)檢測(cè)技術(shù)領(lǐng)域,本發(fā)明為解決現(xiàn)有無人機(jī)目標(biāo)檢測(cè)算法對(duì)于小目標(biāo)檢測(cè)性能差的問題。本發(fā)明包括:建立數(shù)據(jù)集;建立網(wǎng)絡(luò)結(jié)構(gòu):采用多頭自注意力機(jī)制建立多頭自注意力目標(biāo)檢測(cè)頭網(wǎng)絡(luò),所述多頭自注意力目標(biāo)檢測(cè)頭網(wǎng)絡(luò)的后端采用Faster Rcnn基本框架,在多頭自注意力目標(biāo)檢測(cè)頭網(wǎng)絡(luò)的回歸層再次引入自注意力機(jī)制;分步驟進(jìn)行網(wǎng)絡(luò)訓(xùn)練;對(duì)目標(biāo)物體進(jìn)行檢測(cè):對(duì)圖像進(jìn)行預(yù)處理后輸入多頭自注意力目標(biāo)檢測(cè)頭網(wǎng)絡(luò),多頭自注意力目標(biāo)檢測(cè)頭網(wǎng)絡(luò)輸出檢測(cè)結(jié)果。本發(fā)明用于對(duì)大小變化目標(biāo)無人機(jī)的目標(biāo)檢測(cè)。
聲明:
“一種基于多頭自注意力機(jī)制的無人機(jī)變尺寸目標(biāo)檢測(cè)方法” 該技術(shù)專利(論文)所有權(quán)利歸屬于技術(shù)(論文)所有人。僅供學(xué)習(xí)研究,如用于商業(yè)用途,請(qǐng)聯(lián)系該技術(shù)所有人。
我是此專利(論文)的發(fā)明人(作者)