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

有效

• 申請號: 202010084625.6
• 申請日: 2020-02-10
• 公開（公告）號: CN111340819B
• 公開（公告）日: 2023-09-12
• 發明（設計）人: 柳露艷;馬鍇;鄭冶楓
• 申請（專利權）人: 騰訊科技（深圳）有限公司
• 主分類號: G06T7/11
• 分類號: G06T7/11;G06T5/00;G06N3/088;G06N3/0464
• 代理公司: 深圳翼盛智成知識產權事務所(普通合伙) 44300
• 代理人: 蔡艾瑩

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

獲取目標域圖像、以及已標注目標信息的源域圖像；

采用第一生成對抗網絡中的生成網絡分別對源域圖像和目標域圖像進行分割，確定第一源域分割損失和第一目標域分割損失；

采用第二生成對抗網絡中的生成網絡分別對源域圖像和目標域圖像進行分割，確定第二源域分割損失和第二目標域分割損失；

根據所述第一源域分割損失、所述第二源域分割損失確定第一源域目標圖像和第二源域目標圖像，以及根據所述第一目標域分割損失、所述第二目標域分割損失確定第一目標域目標圖像和第二目標域目標圖像；

利用所述第一源域目標圖像以及所述第一目標域目標圖像對所述第二生成對抗網絡進行訓練，利用所述第二源域目標圖像以及所述第二目標域目標圖像對所述第一生成對抗網絡進行訓練，得到訓練后的第一生成對抗網絡；

基于所述訓練后的第一生成對抗網絡的生成網絡對待分割圖像進行分割，得到分割結果。

2.根據權利要求1所述的方法，其特征在于，所述采用第一生成對抗網絡中的生成網絡分別對源域圖像和目標域圖像進行分割，確定第一源域分割損失和第一目標域分割損失，包括：

采用第一生成對抗網絡中的生成網絡分別對源域圖像和目標域圖像進行特征提取，得到第一源域圖像的特征信息和第一目標域圖像的特征信息；

基于所述第一源域圖像的特征信息，對所述源域圖像進行目標分割，確定第一源域分割損失；

基于所述第一目標域圖像的特征信息，對所述目標域圖像進行目標分割，確定第一目標域分割損失。

3.根據權利要求2所述的方法，其特征在于，所述源域圖像包括噪聲圖像和無噪聲圖像，所述基于所述第一源域圖像的特征信息，對所述源域圖像進行目標分割，確定第一源域分割損失，包括：

基于所述第一源域圖像的特征信息，對所述源域圖像中噪聲圖像進行目標分割，得到第一噪聲分割概率；

獲取源域圖像中噪聲圖像的權重圖；

根據所述第一噪聲分割概率，和噪聲圖像的權重圖，獲取第一噪聲分割損失；

基于所述第一源域圖像的特征信息，對所述源域圖像中無噪聲圖像進行目標分割，得到第一無噪聲分割概率；

根據第一無噪聲分割概率，和無噪聲圖像的標注結果，獲取第一無噪聲分割損失；

基于所述第一噪聲分割損失和所述第一無噪聲分割損失，確定第一源域分割損失。

4.根據權利要求2所述的方法，其特征在于，基于所述第一目標域圖像的特征信息，對所述目標域圖像進行目標分割，確定第一目標域分割損失，包括：

基于所述第一目標域圖像的特征信息，對所述目標域圖像進行目標分割，得到第一目標域分割概率；

根據所述第一目標域分割概率，生成第一目標域分割結果；

根據所述第一目標域分割結果、和所述目標域圖像，獲取第一目標域分割損失。

5.根據權利要求1所述的方法，其特征在于，所述采用第二生成對抗網絡中的生成網絡分別對源域圖像和目標域圖像進行分割，確定第二源域分割損失和第二目標域分割損失，包括：

采用第二生成對抗網絡中的生成網絡分別對源域圖像和目標域圖像進行特征提取，得到第二源域圖像的特征信息和第二目標域圖像的特征信息；

基于所述第二源域圖像的特征信息，對所述源域圖像進行目標分割，確定第二源域分割損失；

基于所述第二目標域圖像的特征信息，對所述目標域圖像進行目標分割，確定第二目標域分割損失。

6.根據權利要求1所述的方法，其特征在于，所述根據所述第一源域分割損失、所述第二源域分割損失確定第一源域目標圖像和第二源域目標圖像，包括：

對所述第一源域分割損失進行排序，從排序后的第一源域分割損失中選取滿足預設損失條件的源域圖像，確定為第一源域目標圖像；

對所述第二源域分割損失進行排序，從排序后的第二源域分割損失中選取滿足預設損失條件的源域圖像，確定為第二源域目標圖像。