[發明專利]圖像分割方法、裝置、電子設備及可讀存儲介質

有效

• 申請號: 202111042037.7
• 申請日: 2021-09-07
• 公開（公告）號: CN113516672B
• 公開（公告）日: 2022-02-25
• 發明（設計）人: 李博賢;周朋;陶穎;鄭鵬程
• 申請（專利權）人: 北京美攝網絡科技有限公司
• 主分類號: G06T7/12
• 分類號: G06T7/12;G06T7/13
• 代理公司: 北京潤澤恒知識產權代理有限公司 11319
• 代理人: 莎日娜

本申請提供了一種圖像分割方法、裝置、電子設備及可讀存儲介質。所述方法包括：以視頻流中的每一幀圖像為當前幀圖像，確定當前幀圖像是檢測幀圖像還是跟蹤幀圖像；在當前幀圖像是跟蹤幀圖像的情況下，根據為當前幀圖像預先存儲的參考裁剪區域和用于框選當前幀圖像的前一幀圖像中目標物的外接矩形，確定當前幀圖像的裁剪區域；根據當前幀圖像的裁剪區域，對當前幀圖像進行裁剪，得到當前幀圖像的局部圖像；將當前幀圖像的局部圖像輸入目標物分割模型，得到當前幀圖像的局部圖像對應的掩膜并從當前幀圖像中分割出目標物。通過本申請的方法，可通過逐幀跟蹤增大有效目標物在輸入模型的圖像中的占比來提升精度，以實現圖像分割速度與精度的兼得。

技術領域

本申請實施例涉及圖像處理領域，尤其涉及一種圖像分割方法、裝置、電子設備及可讀存儲介質。

背景技術

在當前快節奏生活的影響下，隨著技術的發展，短視頻與直播內容的量越來越多，涉及到的領域也越來越廣泛，基于視頻的目標物分割需求也愈來愈多。目標物分割任務通常用于目標物摳圖、直播背景替換、智能攝像等領域。對于視頻分割任務，對圖像處理的穩定性和處理速度都有很高的要求，而當前傳統的圖像分割方法無法同時兼顧速度與精度。

發明內容

本申請提供一種圖像分割方法、裝置、電子設備及可讀存儲介質，通過逐幀跟蹤增大有效目標物在輸入模型的圖像中的占比來提升精度，從而實現圖像分割速度與精度的兼得。

本申請實施例第一方面提供了一種圖像分割方法，所述方法包括：

以視頻流中的每一幀圖像為當前幀圖像，確定所述當前幀圖像是檢測幀圖像還是跟蹤幀圖像；

在所述當前幀圖像是跟蹤幀圖像的情況下，根據為所述當前幀圖像預先存儲的參考裁剪區域和用于框選所述當前幀圖像的前一幀圖像中目標物的外接矩形，確定所述當前幀圖像的裁剪區域；其中，所述為所述當前幀圖像預先存儲的參考裁剪區域是所述當前幀圖像的前一幀圖像的裁剪區域；

根據所述當前幀圖像的裁剪區域，對所述當前幀圖像進行裁剪，得到所述當前幀圖像的局部圖像；

將所述當前幀圖像的局部圖像輸入目標物分割模型，得到所述當前幀圖像的局部圖像對應的掩膜；

根據所述當前幀圖像的局部圖像對應的掩膜，從所述當前幀圖像中分割出所述目標物。

可選的，所述方法還包括：

在所述當前幀圖像是視頻流中的第一幀圖像的情況下，或，在所述當前幀圖像及所述當前幀圖像的前一幀圖像均為檢測幀圖像的情況下，

將所述當前幀圖像輸入所述目標物分割模型，得到所述當前幀圖像對應的掩膜；

根據所述當前幀圖像中目標物所在的區域，生成用于框選所述當前幀圖像中目標物的外接矩形；

根據所述用于框選所述當前幀圖像中目標物的外接矩形，得到所述當前幀圖像的裁剪區域；

存儲所述當前幀圖像的裁剪區域，以作為所述當前幀圖像的下一幀圖像的參考裁剪區域。

可選的，所述方法還包括：

在所述當前幀圖像為檢測幀圖像，且所述當前幀圖像的前一幀圖像為跟蹤幀圖像情況下，

將所述當前幀圖像輸入所述目標物分割模型，得到所述當前幀圖像對應的掩膜；

根據所述當前幀圖像對應的掩膜，確定所述當前幀圖像中目標物所在的區域；

將所述當前幀圖像的前一幀圖像中目標物所在的區域的有效像素的數量與所述當前幀圖像中目標物所在的區域的有效像素的數量進行對比，確定是否滿足滯留條件；