[發明專利]網絡模型的訓練方法及裝置、標注信息的確定方法及裝置

有效

• 申請號: 202010961988.3
• 申請日: 2020-09-14
• 公開（公告）號: CN112101453B
• 公開（公告）日: 2021-11-02
• 發明（設計）人: 于朋鑫;夏晨;張榮國;李新陽;王少康;陳寬
• 申請（專利權）人: 推想醫療科技股份有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06K9/46;G06T5/00
• 代理公司: 北京布瑞知識產權代理有限公司 11505
• 代理人: 孟潭

1.一種網絡模型的訓練方法，其特征在于，包括：

確定初始網絡模型；

確定待標注樣本圖像對應的第一標注信息；

基于所述第一標注信息和所述待標注樣本圖像確定所述待標注樣本圖像對應的概率圖像；

基于待標注樣本圖像和所述待標注樣本圖像對應的概率圖像訓練所述初始網絡模型，以生成圖像理解模型，其中，所述圖像理解模型用于基于待標注圖像生成所述待標注圖像對應的特征圖和修復圖像，所述概率圖像用于表征所述待標注樣本圖像中的顯示單元被擦除的概率；

其中，所述初始網絡模型包括特征提取模塊和圖像修復模塊，所述基于待標注樣本圖像和所述待標注樣本圖像對應的概率圖像訓練所述初始網絡模型，包括：

基于所述待標注樣本圖像和所述概率圖像生成所述待標注樣本圖像對應的擦除圖像；

將所述擦除圖像輸入所述特征提取模塊，以生成特征圖；

將所述特征圖輸入所述圖像修復模塊，以生成修復圖像；

基于所述待標注樣本圖像和所述修復圖像訓練所述初始網絡模型。

2.根據權利要求1所述的方法，其特征在于，所述基于所述待標注樣本圖像和所述概率圖像生成所述待標注樣本圖像對應的擦除圖像，包括：

基于所述概率圖像對所述待標注樣本圖像進行擦除操作，以生成第一階段擦除圖像，其中，所述第一階段擦除圖像包括第一擦除區域；

將所述第一階段擦除圖像確定為所述擦除圖像。

3.根據權利要求2所述的方法，其特征在于，所述第一擦除區域的數量為多個，在所述將所述第一階段擦除圖像確定為所述擦除圖像之前，還包括：

基于所述多個第一擦除區域各自對應的尺寸信息和預設尺寸條件生成第一擦除區域集合；

基于所述第一擦除區域集合中的第一擦除區域之間的距離信息，確定第二階段擦除圖像；

其中，所述將所述第一階段擦除圖像確定為所述擦除圖像，包括：

將所述第二階段擦除圖像確定為所述擦除圖像。

4.根據權利要求3所述的方法，其特征在于，所述基于所述第一擦除區域集合中的第一擦除區域之間的距離信息，確定第二階段擦除圖像，包括：

針對所述第一擦除區域集合中的任意兩個第一擦除區域，計算所述兩個第一擦除區域之間的歐式距離；

如果所述歐式距離符合預設連通條件，則將所述兩個第一擦除區域進行連通處理；

基于經所述連通處理后的所述第一擦除區域集合生成所述第二階段擦除圖像。

5.根據權利要求1所述的方法，其特征在于，所述基于所述第一標注信息和所述待標注樣本圖像確定所述概率圖像，包括：

基于所述第一標注信息確定所述待標注樣本圖像的第一區域和第二區域，其中，所述第一區域包括多個第一顯示單元，所述第二區域包括多個第二顯示單元；

針對所述多個第一顯示單元，分別計算所述第一顯示單元和與所述第一顯示單元最接近的第二顯示單元之間的第一歐式距離，以生成第一距離集合；

針對所述多個第二顯示單元，分別計算所述第二顯示單元和與所述第二顯示單元最接近的第一顯示單元之間的第二歐式距離，以生成第二距離集合；

基于所述第一距離集合和所述第二距離集合生成所述概率圖像。

6.根據權利要求5所述的方法，其特征在于，所述第一距離集合包括多個第一歐式距離，所述第二距離集合包括多個第二歐式距離，所述基于所述第一距離集合和所述第二距離集合生成所述概率圖像，包括：

基于所述多個第一歐式距離確定預設距離閾值；

基于所述多個第一歐式距離與所述預設距離閾值的關系，確定所述多個第一顯示單元各自對應的擦除概率；

基于所述多個第二歐式距離與所述預設距離閾值的關系，確定所述多個第二顯示單元各自對應的擦除概率；

基于所述多個第一顯示單元各自對應的擦除概率和所述多個第二顯示單元各自對應的擦除概率，生成所述概率圖像。