[發明專利]圖像標注方法、裝置、計算機設備及存儲介質

有效

• 申請號: 202010436238.4
• 申請日: 2020-05-21
• 公開（公告）號: CN111612068B
• 公開（公告）日: 2023-01-06
• 發明（設計）人: 曹世磊;王淑欣;謝蓯;王連生;鄭冶楓
• 申請（專利權）人: 騰訊科技（深圳）有限公司
• 主分類號: G06V10/764
• 分類號: G06V10/764;G06V10/774;G06N3/04;G06N3/08
• 代理公司: 北京三高永信知識產權代理有限責任公司 11138
• 代理人: 張所明

1.一種圖像標注方法，其特征在于，所述方法包括：

獲取第一樣本圖像和第二樣本圖像，所述第一樣本圖像和所述第二樣本圖像為腹部圖像；調用圖像處理模型，對所述第一樣本圖像和所述第二樣本圖像進行對比處理，得到由所述第一樣本圖像映射到所述第二樣本圖像的第一樣本映射信息，以及由所述第二樣本圖像映射到所述第一樣本圖像的第二樣本映射信息，所述第一樣本映射信息包括多個位置對應的第一映射值，所述第二樣本映射信息包括所述多個位置對應的第二映射值；根據所述第一樣本映射信息對所述第一樣本圖像進行映射處理，得到第一重建圖像；獲取所述第二樣本圖像和所述第一重建圖像之間的第一相似值；根據所述第一樣本圖像中每個位置對應的第一映射值，以及所述第二樣本圖像中每個位置對應的第二映射值，獲取所述第一樣本圖像中每個位置對應的第一差異值；根據每個位置對應的所述第一差異值和所述第一相似值，獲取第一相似性損失值，所述第一相似性損失值表示所述第一重建圖像與所述第二樣本圖像之間的相似程度；根據所述第一相似性損失值，訓練所述圖像處理模型；

獲取目標腹部圖像、模板腹部圖像和所述模板腹部圖像對應的模板標注圖像，所述模板標注圖像包括所述模板腹部圖像中的脾、左腎、食道、肝、胃、十二指腸和胰腺的標注標簽；

調用所述圖像處理模型中的第一編碼網絡中的n個編碼層，對所述模板腹部圖像依次進行編碼處理，得到所述模板腹部圖像對應的模板特征信息，所述n為大于1的正整數；調用所述圖像處理模型中的第二編碼網絡中的n個編碼層，對所述目標腹部圖像依次進行編碼處理，得到所述目標腹部圖像對應的目標特征信息；所述第一編碼網絡和所述第二編碼網絡是所述圖像處理模型中的兩個相同的并行分支；

調用所述圖像處理模型中的解碼網絡中的第一個解碼層中的第一對比單元，對所述模板特征信息和所述目標特征信息進行對比處理，得到所述目標特征信息與所述模板特征信息之間的差異信息；調用所述第一個解碼層中的第一解碼單元中的多個卷積單元，對所述差異信息和所述目標特征信息進行解碼處理，得到映射信息；

調用所述解碼網絡中的第二個解碼層中的第一映射單元，對所述第二個解碼層對應的編碼層輸出的模板特征信息和所述第一個解碼層輸出的映射信息進行映射處理，得到重建特征信息；調用所述第二個解碼層中的第二對比單元，對所述第二個解碼層對應的編碼層輸出的目標特征信息和所述重建特征信息進行對比處理，得到差異信息；調用所述第二個解碼層中的第二解碼單元中的多個卷積單元，對所述第二個解碼層對應的編碼層輸出的目標特征信息、所述差異信息和所述第一個解碼層輸出的映射信息進行解碼處理，得到下一個映射信息，直至得到所述解碼網絡中的最后一個解碼層輸出的映射信息；其中，所述解碼網絡包括n個解碼層，第k個解碼層與第n+1-k個編碼層對應，所述k為不大于n的正整數；

根據最后一個解碼層輸出的映射信息對所述模板標注圖像進行映射處理，得到所述目標腹部圖像對應的目標標注圖像，所述目標標注圖像包括所述目標腹部圖像中的脾、左腎、食道、肝、胃、十二指腸和胰腺的標注標簽。

2.根據權利要求1所述的方法，其特征在于，所述獲取目標腹部圖像、模板腹部圖像和所述模板腹部圖像對應的模板標注圖像，包括：

獲取所述目標腹部圖像、訓練所述圖像處理模型時采用的多個樣本模板腹部圖像和所述多個樣本模板腹部圖像對應的樣本模板標注圖像；

調用所述圖像處理模型，分別對所述多個樣本模板腹部圖像和所述目標腹部圖像進行對比處理，得到由每個樣本模板腹部圖像映射到所述目標腹部圖像的第一樣本映射信息；

分別根據每個第一樣本映射信息，對所述每個第一樣本映射信息對應的樣本模板腹部圖像進行映射處理，得到所述每個樣本模板腹部圖像對應的重建模板腹部圖像；

從多個重建模板腹部圖像中確定與所述目標腹部圖像的關聯度最高的重建模板腹部圖像，選取所述重建模板腹部圖像對應的樣本模板腹部圖像以及樣本模板標注圖像。