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

基本信息

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

摘要

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