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

有效

• 申請號: 201810576571.8
• 申請日: 2018-06-06
• 公開（公告）號: CN110570434B
• 公開（公告）日: 2022-09-30
• 發明（設計）人: 周健;孫海鳴;謝迪;浦世亮
• 申請（專利權）人: 杭州海康威視數字技術股份有限公司
• 主分類號: G06T7/12
• 分類號: G06T7/12;G06T7/136;G06T7/149;G06T7/194;G06N3/04;G06N3/08
• 代理公司: 北京博思佳知識產權代理有限公司 11415
• 代理人: 林祥

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

從待分割圖像中截取圖像塊，所述圖像塊中包含一個待分割的目標對象；

使用訓練好的分割模型對所述圖像塊進行分割，得到所述目標對象的分割結果；所述分割模型用于預測不同類別的物體的分割結果并為不同類別的物體的分割結果賦予相同的真值；真值用于標識物體的類別；

接收標注指令，將所述目標對象的分割結果的真值設置成所述標注指令所指定的真值；

其中，所述分割模型通過以下方式訓練得到：

獲取訓練數據集，所述訓練數據集包括原始圖片和與原始圖片對應的標簽圖片；原始圖片中僅包含一個物體，標簽圖片上標記有物體的分割結果和該分割結果的真值，每張標簽圖片上標記的真值相同；

利用所述訓練數據集對通用的分割模型進行一定數量的訓練后，得到訓練好的分割模型。

2.如權利要求1所述的方法，其特征在于，所述從待分割圖像中截取圖像塊，包括：

在所述待分割圖像中確定一個待分割的目標對象，所述目標對象通過矩形框標記出；

根據預設的外擴系數，將所述目標對象的矩形框向外擴張得到新矩形框，所述新矩形框完全包含所述目標對象并靠近所述目標對象的邊緣；

將所述新矩形框所包圍的圖像塊從待分割圖像中截取出。

3.如權利要求2所述的方法，其特征在于，若外擴前的矩形框的左上頂點為(x，y)，長為L，寬為W；預設的外擴系數為r；則

外擴后的所述新矩形框的長L_new滿足：L_new＝L*(1+r)；

外擴后的所述新矩形框的寬W_new滿足：W_new＝W*(1+r)；

外擴后的所述新矩形框的左上頂點(x_new，y_new)滿足：

4.如權利要求1所述的方法，其特征在于，在使用訓練好的分割模型對所述圖像塊進行分割，得到所述目標對象的分割結果之后，所述方法還包括：

將所述目標對象的分割結果疊加到所述待分割圖像中的所述目標對象的位置處。

5.如權利要求1所述的方法，其特征在于，在使用訓練好的分割模型對所述圖像塊進行分割，得到所述目標對象的分割結果之后，所述方法還包括：

從所述目標對象的分割結果中提取所述目標對象的輪廓；

對所述輪廓進行多邊形擬合，得到包含有限個輪廓點的輪廓擬合線；

針對所述輪廓擬合線中的每兩個相鄰輪廓點，計算該兩個相鄰輪廓點之間的距離，若該距離超過設定閾值則在該兩個相鄰輪廓點間增加新的輪廓點；

當輪廓擬合線中任兩個相鄰輪廓點間的距離均不超過設定閾值時，將該輪廓擬合線疊加到所述待分割圖像中的所述目標對象的位置處，所述輪廓擬合線圍住的區域為目標對象的分割結果。

6.如權利要求5所述的方法，其特征在于，所述計算該兩個相鄰輪廓點之間的距離，若該距離超過設定閾值則在該兩個相鄰輪廓點間增加新的輪廓點，包括：

若該兩個相鄰輪廓點的坐標分別為(x_a，y_a)和(x_b，y_b)，則計算得到該兩個相鄰輪廓點間的距離

若該兩個相鄰輪廓點間的距離d_ab大于設定閾值Distance_Threshold，則在該兩個相鄰輪廓點間增加n_ab個新的輪廓點；

其中，n_ab＝floor(d_ab/Distance_Threshold)-1；

1≤k≤n_ab-1，第k個新的輪廓點的坐標(x_k，y_k)滿足：

7.如權利要求5所述的方法，其特征在于，在將該輪廓擬合線疊加到所述待分割圖像中的所述目標對象的位置處之后，所述方法還包括：

接收針對該輪廓擬合線中的輪廓點的拖曳指令；

根據所述拖曳指令調整輪廓點的位置。