[發明專利]一種圖像分割方法及裝置

有效

• 申請號: 201910075286.2
• 申請日: 2019-01-25
• 公開（公告）號: CN111489359B
• 公開（公告）日: 2023-05-30
• 發明（設計）人: 黃永禎;曹春水;王繼隆
• 申請（專利權）人: 銀河水滴科技(北京)有限公司
• 主分類號: G06T7/11
• 分類號: G06T7/11
• 代理公司: 北京超成律師事務所 11646
• 代理人: 胡上海

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

將獲取的模板圖像和待分割圖像輸入至預先訓練好的分割模型中，執行以下處理：

對所述模板圖像經過不同次卷積處理后得到不同維度的第一特征圖像；

將所述待分割圖像經過第n次卷積處理后得到的特征圖像與第n維第一特征圖像進行拼接，將拼接后的圖像進行卷積處理后得到所述待分割圖像的第n維第二特征圖像；

在確定第n維之后的第二特征圖像時，依次將第n+i-1維第二特征圖像與第n+i-1維的第一特征圖像進行拼接，將拼接后得到的圖像進行卷積后得到所述待分割圖像的第n+i維第二特征圖像，直至確定出所述待分割圖像的第m維第二特征圖像；

基于確定出的第m維第二特征圖像，確定所述待分割圖像的分割圖像；

其中，n、m為正整數，m大于n，i取遍1至m-n中的整數，按照下列方式訓練所述分割模型：

獲取樣本圖像集合，其中，所述樣本圖像集合包括多個樣本圖像、與每個樣本圖像對應的樣本模板圖像和參考分割圖像；

從所述樣本圖像集合中選取樣本圖像、與所述樣本圖像對應的樣本模板圖像輸入至待訓練的分割模型中，執行以下訓練過程，直至確定所述待訓練的分割模型訓練完成，其中，訓練過程包括：

對所述樣本圖像對應的樣本模板圖像經過不同次卷積處理后得到不同維度的第一樣本特征圖像；

將所述樣本圖像經過第n次卷積處理后得到的特征圖像與第n維第一樣本特征圖像進行拼接，將拼接后的圖像進行卷積處理后得到所述樣本圖像的第n維第二樣本特征圖像；

在確定第n維之后的第二樣本特征圖像時，依次將第n+i-1維第二樣本特征圖像與第n+i-1維的第一樣本特征圖像進行拼接，將拼接后得到的圖像進行卷積后得到所述樣本圖像的第n+i維第二樣本特征圖像，直至確定出所述樣本圖像的第m維第二樣本特征圖像；

基于確定出的第m維第二樣本特征圖像，確定所述樣本圖像的分割圖像；其中，n、m為正整數，m大于n，i取遍1至m-n中的整數；

基于所述樣本圖像的分割圖像以及所述樣本圖像對應的參考分割圖像，確定本次訓練過程的損失值；

當所述損失值大于預設損失值時，調整所述待訓練的分割模型，并從所述樣本圖像集合中繼續選取樣本圖像和選取的樣本圖像對應的樣本模板圖像輸入至調整后的待訓練的分割模型中，再次執行所述訓練過程；

當所述損失值小于或等于所述預設損失值時，確定所述待訓練的分割模型訓練完成。

2.根據權利要求1所述的方法，其特征在于，所述基于確定出的第m維第二特征圖像，確定所述待分割圖像的分割圖像，包括：

將所述第m維第二特征圖像進行上采樣處理，并根據經過上采樣處理之后的第m維第二特征圖像中每個像素點的像素值，確定所述待分割圖像的分割圖像。

3.根據權利要求2所述的方法，其特征在于，所述根據經過上采樣處理之后的第m維第二特征圖像中每個像素點的像素值，確定所述待分割圖像的分割圖像，包括：

將所述經過上采樣處理之后的第m維第二特征圖像中像素值大于預設像素值的像素點劃分為第一類像素點，所述第一類像素點為所述待分割圖像與所述模板圖像具有相同圖像特征的像素點；

將所述經過上采樣處理之后的第m維第二特征圖像中像素值不大于所述預設像素值的像素點劃分為第二類像素點，所述第二類像素點為所述待分割圖像與所述模板圖像的圖像特征存在差異的像素點；

基于所述第一類像素點和所述第二類像素點，確定所述待分割圖像的分割圖像。

4.根據權利要求3所述的方法，其特征在于，所述基于所述第一類像素點和所述第二類像素點，確定所述待分割圖像的分割圖像，包括：

將所述第一類像素點的像素值均調整為第一數值，以及，將所述第二類像素點的像素值均調整為第二數值；

將所述待分割圖像劃分為像素值為所述第一數值的第一圖像分割區域、以及像素值為所述第二數值的第二圖像分割區域。