[發明專利]一種目標提取方法及裝置

在審

• 申請號: 202010952509.1
• 申請日: 2020-09-11
• 公開（公告）號: CN112070793A
• 公開（公告）日: 2020-12-11
• 發明（設計）人: 王曉茹;徐培容;曲昭偉;張珩;谷嘉航
• 申請（專利權）人: 北京郵電大學
• 主分類號: G06T7/194
• 分類號: G06T7/194;G06T7/66;G06K9/38;G06K9/62;G06N3/04
• 代理公司: 北京集佳知識產權代理有限公司 11227
• 代理人: 尹秀

本申請提供了一種目標提取方法及裝置，其中，方法包括：獲取待提取目標的圖像和位置信息；位置信息包括前景點的位置信息和背景點的位置信息；依據前景點的位置信息計算前景二值圖像，以及依據背景點的位置信息計算背景二值圖像；將圖像、前景二值圖像和背景二值圖像進行通道組合；將通道組合后的圖像輸入訓練后的目標語義分割模型，得到用于區分所述圖像中的目標和非目標的掩膜；目標語義分割模型是至少將Deeplab v3+語義分割網絡中的Xception?65替換為殘差網絡ResNet?101得到；依據掩膜和圖像，提取圖像中的目標。本申請提供的目標的提取方法，提取目標的速度和精度都得到提高。

技術領域

本申請涉及圖像處理領域，尤其涉及一種目標提取方法及裝置。

背景技術

在一些應用場景中，用戶需要從圖像中提取需要的物體，為了描述方便，將圖像中用戶所需提取的物體稱為目標，將圖像中除目標外的內容，稱為非目標。通常情況下，用戶在圖像中的目標內點擊預設第一數量的位置點(為了描述方便，稱為前景點)，該前景點用于代表目標。同時，用戶在圖像的非目標點擊預設的第二數量的位置點(為了描述方便，稱為背景點)，該背景點用于代表非目標。

目前，基于待提取目標的圖像、前景點和背景點，通過語義分割模型進行處理，進而提取出目標。

但是，目前提取目標的精度和速度都較低。

發明內容

本申請提供了一種目標提取方法及裝置，目的在于解決目標提取的精度和速度較低的問題。

為了實現上述目的，本申請提供了以下技術方案：

本申請提供了一種目標提取方法，包括：

獲取待提取目標的圖像和位置信息；所述位置信息包括前景點的位置信息和背景點的位置信息；

依據所述前景點的位置信息計算前景二值圖像；

依據所述背景點的位置信息計算背景二值圖像；

將所述圖像、所述前景二值圖像和所述背景二值圖像進行通道組合，得到通道組合后的圖像；

將所述通道組合后的圖像輸入訓練后的目標語義分割模型，得到用于區分所述圖像中的目標和非目標的掩膜；所述目標語義分割模型是至少將Deeplab v3+語義分割網絡中的Xception-65替換為殘差網絡ResNet-101得到；

依據所述掩膜和所述圖像，提取所述圖像中的目標。

可選的，所述依據所述前景點的位置信息計算前景二值圖像，包括：

在所述圖像中，分別以每個前景點為圓心，以預設半徑作圓，得到每個所述前景點對應的圓；

將每個所述前景點對應的圓所在的區域的像素點的像素值設置為1，將所述前景點對應的圓之外的像素點的像素值設置為0，得到所述前景二值圖像。

可選的，所述依據所述背景點的位置信息計算背景二值圖像，包括：

在所述圖像中，分別以每個背景點為圓心，以所述預設半徑作圓，得到每個所述背景點對應的圓；

將每個所述背景點對應的圓所在的區域的像素點的像素值設置為1，將所述背景點對應的圓之外的像素點的像素值設置為0，得到所述背景二值圖像。

可選的，所述目標語義分割網絡的最終輸出網絡的通道數為2；

所述掩膜包括第一掩膜和第二掩膜；所述第一掩膜中任一位置點的取值表示所述圖像中與該位置點相同的像素點屬于所述目標的概率；所述第二掩膜中任一位置點的取值表示所述圖像中與該位置點相同的像素點屬于所述非目標的概率。

所述依據所述掩膜和所述圖像，提取所述圖像中的所述目標，包括：