[發明專利]圖像分割方法、裝置、設備及存儲介質

在審

• 申請號: 202110558675.8
• 申請日: 2021-05-21
• 公開（公告）號: CN113159056A
• 公開（公告）日: 2021-07-23
• 發明（設計）人: 李陽;吳劍煌
• 申請（專利權）人: 中國科學院深圳先進技術研究院
• 主分類號: G06K9/34
• 分類號: G06K9/34;G06K9/62;G06N3/08
• 代理公司: 北京品源專利代理有限公司 11332
• 代理人: 潘登

1.一種圖像分割方法，其特征在于，包括：

獲取至少一張待分割圖像；

將所述待分割圖像輸入至預先訓練完成的圖像分割模型中，得到與所述待分割圖像對應的目標分割圖像；

其中，所述圖像分割模型基于編碼器、解碼器以及至少一個自注意力模型構建，所述自注意力模型用于確定所述待分割圖像中每個像素點與圖像中所有像素點之間的依賴關系。

2.根據權利要求1所述的方法，其特征在于，所述將所述待分割圖像輸入至預先訓練完成的圖像分割模型中，得到與所述待分割圖像對應的目標分割圖像，包括：

將所述待分割圖像輸入至預先訓練完成的編碼器中，得到與所述待分割圖像對應的目標編碼圖像；

將所述目標編碼圖像輸入至預先訓練完成的至少一個自注意力模型中，得到與所述目標編碼圖像對應的自注意力分割圖像；

將所述自注意力分割圖像輸入至預先訓練完成的解碼器中，得到與所述待分割圖像對應的目標分割圖像。

3.根據權利要求2所述的方法，其特征在于，如果所述目標編碼圖像為平面圖像，所述圖像分割模型包括第一轉換層和第二轉化層；

在所述得到與所述待分割圖像對應的目標編碼圖像之后，所述將所述目標編碼圖像輸入至預先訓練完成的至少一個自注意力模型中之前，還包括：

將所述目標編碼圖像輸入至所述第一轉換層，以將所述目標編碼圖像由二維圖像特征轉化成一維圖像特征；

在所述將所述自注意力分割圖像輸入至預先訓練完成的解碼器中之前，還包括：

將所述自注意力分割圖像輸入至所述第二轉換層，以將所述自注意力分割圖像由一維圖像特征轉化成二維圖像特征。

4.根據權利要求1所述的方法，其特征在于，還包括：

基于多組訓練樣本數據，對基于預先建立的初始網絡模型進行訓練，生成圖像分割模型，其中，所述訓練樣本數據包括樣本圖像數據以及與所述樣本待分割圖像對應的樣本目標分割圖像。

5.根據權利要求4所述的方法，其特征在于，所述基于多組訓練樣本數據，對基于預先建立的初始網絡模型進行訓練，包括：

將所述樣本圖像數據輸入至預先建立的編碼器中，得到與所述待分割圖像對應的樣本編碼圖像；

將所述樣本編碼圖像輸入至預先建立的至少一個自注意力模型中，得到與所述目標編碼圖像對應的樣本自注意力圖像；

將所述樣本自注意力圖像輸入至預先建立的解碼器中，得到與所述待分割圖像對應的目標分割圖像。

6.根據權利要求5所述的方法，其特征在于，所述將所述樣本編碼圖像輸入至預先建立的至少一個自注意力模型中，得到與所述目標編碼圖像對應的樣本自注意力圖像，包括：

將所述樣本編碼圖像輸入至預先建立的自注意力模型中；

基于所述樣本編碼圖像進行線性變化得到所述自注意力模型的第一待調整參數矩陣、第二待調整參數矩陣和第三待調整參數矩陣；

基于所述第一待調整參數矩陣和所述第二待調整參數矩陣確定與所述樣本編碼圖像對應的相似度矩陣；

基于所述第三待調整參數矩陣對所述相似度矩陣進行加權，得到加權特征圖像；

基于至少兩張加權特征圖像和所述樣本編碼圖像確定與所述目標編碼圖像對應的樣本自注意力圖像。

7.根據權利要求6所述的方法，其特征在于，所述基于所述第一待調整參數矩陣和所述第二待調整參數矩陣確定與所述樣本編碼圖像對應的相似度矩陣，包括：

將所述樣本編碼圖像中每一個像素點逐個確定為目標像素點；

針對每個所述目標像素點，基于所述第一待調整參數矩陣和所述第二待調整參數矩陣分別計算所述目標像素點與所述樣本編碼圖像中所有像素點之間的像素相似度；

基于每一個所述目標像素點在所述樣本編碼圖像中所處的位置以及各個所述像素相似度構建與所述樣本編碼圖像對應的相似度矩陣。