[發明專利]一種基于目標框標注的實例分割方法

在審

• 申請號: 202011401506.5
• 申請日: 2020-12-03
• 公開（公告）號: CN112508029A
• 公開（公告）日: 2021-03-16
• 發明（設計）人: 許明江
• 申請（專利權）人: 蘇州科本信息技術有限公司
• 主分類號: G06K9/34
• 分類號: G06K9/34;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 北京眾合誠成知識產權代理有限公司 11246
• 代理人: 郭曉鳳

1.一種基于目標框標注的實例分割方法，其特征在于：包括如下步驟，

(1)圖片預處理：對于給定不同尺寸的圖片，首先經過預處理；

(2)實例分割器構建：再將預處理后的圖片送入實例分割器中的FPN網絡，得到不同尺寸的特征，隨后這些特征被送入頭模塊中進行實例級別類別和目標框的預測；同時在實例分割器上有一路掩膜預測分支，掩膜預測分支通過控制動態生成的參數來實現動態生成掩碼,用于預測實例的掩膜向量；

(3)目標檢測及掩膜預測：測試時模型將輸出大量潛在目標，對這些目標按照分類得分進行降序排列，并根據目標框間或是掩膜間的交并比進行非極大值抑制；最終從剩下的候選中選取得分排名前K個目標，其對應的掩膜是各個目標實例的掩膜結果，完成實例分割任務；

(4)損失函數計算：訓練時通過基于目標框標注的實例分割損失函數計算掩膜分支的損失函數,實現梯度的反向傳播，完成模型訓練，損失函數包括投影損失和成對損失兩部分。

2.根據權利要求1所述的一種基于目標框標注的實例分割方法，其特征在于：所述步驟(1)中的預處理是將不同尺寸的圖片按下列公式I∈R^H×W×3進行數據增強，其中H是圖片的高，W是圖片的寬。

3.根據權利要求1所述的一種基于目標框標注的實例分割方法，其特征在于：所述步驟(2)中的實例分割器是利用基于anchor-free的單階段實例分割器CondInst。

4.根據權利要求1所述的一種基于目標框標注的實例分割方法，其特征在于：所述步驟(3)中的非極大值抑制算法進行后處理，能有效抑制大量低質量的預測結果，輸出高質量的實例掩膜。

5.根據權利要求1所述的一種基于目標框標注的實例分割方法，其特征在于：所述步驟(4)中的利用投影損失和成對損失兩個損失函數進行掩膜向量訓練，只使用目標框標注的信息便能夠實現實例分割任務。