一種基于注意力模型的遙感圖像船舶檢測(cè)方法和裝置

基本信息

申請(qǐng)?zhí)?/td> CN202210579628.6 申請(qǐng)日 -
公開（公告）號(hào) CN114677596A 公開（公告）日 2022-06-28
申請(qǐng)公布號(hào) CN114677596A 申請(qǐng)公布日 2022-06-28
分類號(hào) G06V20/10（2022.01）I;G06V20/54（2022.01）I;G06V10/774（2022.01）I;G06V10/82（2022.01）I;G06V10/26（2022.01）I;G06V10/44（2022.01）I;G06N3/04（2006.01）I;G06N3/08（2006.01）I;G06K9/62（2022.01）I 分類 計(jì)算；推算；計(jì)數(shù);
發(fā)明人 黎晨陽(yáng);王軍;徐曉剛;徐冠雷;朱亞光 申請(qǐng)（專利權(quán)）人 浙江工商大學(xué)
代理機(jī)構(gòu) 杭州浙科專利事務(wù)所（普通合伙） 代理人 -
地址 311100浙江省杭州市余杭區(qū)中泰街道之江實(shí)驗(yàn)室南湖總部
法律狀態(tài) -

摘要

摘要 本發(fā)明公開一種基于注意力模型的遙感圖像船舶檢測(cè)方法和裝置，該方法包括：步驟一，收集船舶遙感圖像，使用標(biāo)注信息對(duì)圖像數(shù)據(jù)集進(jìn)行數(shù)據(jù)擴(kuò)充；步驟二，對(duì)收集的遙感圖像進(jìn)行預(yù)處理，得到船舶檢測(cè)模型訓(xùn)練使用的數(shù)據(jù)集；步驟三，將訓(xùn)練數(shù)據(jù)集里的圖像輸入到針對(duì)小尺寸目標(biāo)改進(jìn)過(guò)的YOLOV5注意力模型中，進(jìn)行訓(xùn)練，得到訓(xùn)練好的遙感船舶檢測(cè)模型；步驟四，將待檢測(cè)的遙感圖像裁剪，后輸入到訓(xùn)練好的遙感船舶檢測(cè)模型中，輸出船舶的邊界框和置信度；將所有裁剪圖的邊界框映射回原始遙感圖像，經(jīng)過(guò)置信度閾值過(guò)濾和非極大值抑制過(guò)濾重復(fù)目標(biāo)后，獲得最終的檢測(cè)結(jié)果。本發(fā)明解決了遙感圖像中船舶分布稀疏、尺寸過(guò)小導(dǎo)致的檢測(cè)困難的問(wèn)題。