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

有效

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

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

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

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

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

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

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

其中，所述根據為所述當前幀圖像預先存儲的參考裁剪區域和用于框選所述當前幀圖像的前一幀圖像中目標物的外接矩形，確定所述當前幀圖像的裁剪區域，包括：

將所述當前幀圖像的前一幀圖像的裁剪區域由內而外劃分為三個互不重疊的區域，作為所述當前幀圖像的穩定區、回彈區以及延展區；

根據所述前一幀圖像中的目標物的外接矩形與所述當前幀圖像的穩定區、回彈區以及延展區之間的位置關系，確定所述當前幀圖像的裁剪區域。

2.根據權利要求1所述的方法，其特征在于，所述方法還包括：

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

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

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

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

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

3.根據權利要求1所述的方法，其特征在于，在確定所述當前幀圖像的裁剪區域之后，所述方法還包括：

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

4.根據權利要求3所述的方法，其特征在于，所述方法還包括：

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

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

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

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

在滿足滯留條件的情況下，將所述當前幀圖像的前一幀圖像的裁剪區域內的掩膜作為所述當前幀圖像的參考裁剪區域內的掩膜，且所述當前幀圖像的參考裁剪區域外仍保留基于所述目標物分割模型而得到的掩膜，從而生成所述當前幀圖像的最終掩膜；

根據所述當前幀圖像的最終掩膜，生成用于框選所述當前幀圖像中目標物的外接矩形；

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

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

5.根據權利要求1所述的方法，其特征在于，所述確定所述當前幀圖像是檢測幀圖像還是跟蹤幀圖像，包括：

在所述當前幀圖像是視頻流的第一幀圖像的情況下，確定所述當前幀圖像為檢測幀圖像；

在所述當前幀圖像是所述視頻流的第n幀圖像或第n的倍數幀圖像的情況下，確定所述當前幀圖像為檢測幀圖像，其中，n為預先指定的整數；

在所述當前幀圖像的前一幀圖像中目標物的外接矩形的面積占所述當前幀圖像的前一幀圖像的面積的比例超出預設比例范圍的情況下，確定所述當前幀圖像為檢測幀圖像；

在所述當前幀圖像不是檢測幀圖像的情況下，確定所述當前幀圖像為跟蹤幀圖像。