一種高分辨率遙感圖像實例分割模型的訓練方法及裝置

基本信息

申請?zhí)?/td> CN202110987535.2 申請日 -
公開（公告）號 CN114155247A 公開（公告）日 2022-03-08
申請公布號 CN114155247A 申請公布日 2022-03-08
分類號 G06T7/10（2017.01）I;G06N3/04（2006.01）I;G06N3/08（2006.01）I 分類 計算；推算；計數(shù);
發(fā)明人 張鵬;章一驥;馬天舒;尹鵬飛;潘婷;張瑞祥;楊文 申請（專利權(quán)）人 航天恒星科技有限公司
代理機構(gòu) 中國航天科技專利中心 代理人 臧春喜
地址 100086北京市海淀區(qū)知春路82號院
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種高分辨率遙感圖像實例分割模型的訓練方法及裝置。包括：獲取有標簽的第一遙感圖像和無標簽的第二遙感圖像；將第一遙感圖像和第二遙感圖像輸入至初始高分辨率遙感圖像實例分割模型；基于監(jiān)督分支網(wǎng)絡對第一遙感圖像進行處理，并根據(jù)處理結(jié)果得到監(jiān)督分支網(wǎng)絡的監(jiān)督損失函數(shù)；基于一致性分支網(wǎng)絡對第一遙感圖像和第二遙感圖像進行處理，并根據(jù)處理結(jié)果得到一致性分支網(wǎng)絡的一致性損失函數(shù)；基于監(jiān)督損失函數(shù)和一致性損失函數(shù)，計算得到模型損失函數(shù)；根據(jù)模型損失函數(shù)調(diào)整初始高分辨率遙感圖像實例分割模型的模型參數(shù)，并迭代上述過程，直至得到目標高分辨率遙感圖像實例分割模型。本發(fā)明可以有效地利用未標注數(shù)據(jù)提高模型分割精度。