[發明專利]一種訓練樣本獲得方法、裝置、電子設備和存儲介質

有效

• 申請號: 201910107568.6
• 申請日: 2019-02-02
• 公開（公告）號: CN109753975B
• 公開（公告）日: 2021-03-09
• 發明（設計）人: 徐青松;李青
• 申請（專利權）人: 杭州睿琪軟件有限公司
• 主分類號: G06K9/46
• 分類號: G06K9/46;G06N20/00
• 代理公司: 上海思捷知識產權代理有限公司 31295
• 代理人: 王宏婧

1.一種訓練樣本獲得方法，其特征在于，包括：

獲得視頻中的場景片段；

在所述場景片段中選擇一個包含目標對象的視頻幀作為初始幀，對所述初始幀中所述目標對象所在的目標區域進行標注；

提取所述初始幀中被標注的所述目標區域的特征信息；

以所述初始幀為基準，對所述場景片段中前向和/或后向的視頻幀進行特征搜索，確定各個被搜索幀中特征信息與所述目標區域的特征信息相匹配的區域，并對各個被搜索幀中所確定的區域進行自動標注；如果當前有連續多幀均不存在特征信息與所述目標區域的特征信息相匹配的區域，則先預估所述連續多幀的中間幀的特征點坐標，再依次預估前后幀和所述中間幀的中值幀坐標，直到所述連續多幀全部都完成預估，然后調整標注所述連續多幀中的區域，或者，在預估所述中間幀的特征點坐標后，先調整標注所述中間幀中的區域，然后提取所述中間幀中新標注區域的特征信息，再對前后幀進行目標自動匹配標注處理；

提取所述場景片段中已標注的各個視頻幀的圖像作為訓練樣本。

2.如權利要求1所述的訓練樣本獲得方法，其特征在于，所述獲得視頻中的場景片段，包括：

若所述視頻為單場景視頻，則將所述視頻作為一個場景片段；

若所述視頻為多場景視頻，則利用場景切換檢測技術，將所述視頻劃分為多個場景片段。

3.如權利要求2所述的訓練樣本獲得方法，其特征在于，所述場景切換檢測技術包括：基于像素域的檢測算法、基于壓縮域的檢測算法。

4.如權利要求1所述的訓練樣本獲得方法，其特征在于，在所述提取所述初始幀中被標注的所述目標區域的特征信息之前，還包括：

對所述初始幀進行圖像預處理，以使所述初始幀中所述目標區域的特征信息更加明顯。

5.如權利要求1所述的訓練樣本獲得方法，其特征在于，所述目標區域的特征信息，包括：顏色特征、紋理特征和形狀特征中的一種或多種。

6.如權利要求1所述的訓練樣本獲得方法，其特征在于，所述對所述場景片段中前向和/或后向的視頻幀進行特征搜索，包括：

利用均值漂移算法、Kalman濾波算法或粒子濾波算法，對所述場景片段中前向和/或后向的視頻幀進行特征搜索。

7.如權利要求1所述的訓練樣本獲得方法，其特征在于，所述方法還包括：

如果某一被搜索幀中不存在特征信息與所述目標區域的特征信息相匹配的區域，則獲取目標特征信息，確定該被搜索幀中特征信息與所述目標特征信息相匹配的區域，并對該被搜索幀中所確定的區域進行自動標注；

其中，所述目標特征信息為：該被搜索幀的相鄰預設數量幀中已被標注區域的特征信息。

8.一種訓練樣本獲得裝置，其特征在于，包括：

獲得模塊，用于獲得視頻中的場景片段；

第一標注模塊，用于在所述場景片段中選擇一個包含目標對象的視頻幀作為初始幀，對所述初始幀中的所述目標對象所在的目標區域進行標注；

第一提取模塊，用于提取所述初始幀中被標注的所述目標區域的特征信息；

第二標注模塊，用于以所述初始幀為基準，對所述場景片段中前向和/或后向的視頻幀進行特征搜索，確定各個被搜索幀中特征信息與所述目標區域的特征信息相匹配的區域，并對各個被搜索幀中所確定的區域進行自動標注；如果當前有連續多幀均不存在特征信息與所述目標區域的特征信息相匹配的區域，則先預估所述連續多幀的中間幀的特征點坐標，再依次預估前后幀和所述中間幀的中值幀坐標，直到所述連續多幀全部都完成預估，然后調整標注所述連續多幀中的區域，或者，在預估所述中間幀的特征點坐標后，先調整標注所述中間幀中的區域，然后提取所述中間幀中新標注區域的特征信息，再對前后幀進行目標自動匹配標注處理；

第二提取模塊，用于提取所述場景片段中已標注的各個視頻幀的圖像作為訓練樣本。