[發明專利]基于移動物體檢測的目標檢測自動標注方法及裝置

有效

• 申請號: 201910551489.4
• 申請日: 2019-06-24
• 公開（公告）號: CN110288629B
• 公開（公告）日: 2021-07-06
• 發明（設計）人: 楊文龍
• 申請（專利權）人: 湖北億咖通科技有限公司
• 主分類號: G06T7/20
• 分類號: G06T7/20;G06K9/00
• 代理公司: 北京智匯東方知識產權代理事務所(普通合伙) 11391
• 代理人: 康正德

1.一種基于移動物體檢測的目標檢測自動標注方法，包括：

采用場景分類器確定待檢測視頻的場景類別，選取與所述場景類別具有對應關系的至少一種移動物體檢測算法，所述待檢測視頻包括同一類移動物體；

通過所述至少一種移動物體檢測算法檢測所述待檢測視頻的多幀圖片中的移動物體，對檢測到的所述移動物體標注目標框；

對所述多幀圖片中的移動物體采用預置目標跟蹤算法進行跟蹤，對跟蹤到的移動物體標注目標框；

采用預設算法在所述多幀圖片中移除所述移動物體上重疊的目標框，將剩余目標框作為所述移動物體的最終標注結果；其中

所述采用場景分類器確定待檢測視頻的場景類別，選取與所述場景類別具有對應關系的至少一種移動物體檢測算法，包括如下步驟：

從所述待檢測視頻中選取至少一幀圖片；

將所述至少一幀圖片輸入至所述場景分類器，所述場景分類器依據所述至少一幀圖片對應的場景類別確定所述待檢測視頻的場景類別；

選取與所述場景類別具有對應關系的至少一種移動物體檢測算法，且獲取所述至少一種移動物體檢測算法對應的置信度。

2.根據權利要求1所述的方法，其中，若將從所述待檢測視頻中選取出多幀圖片輸入至所述場景分類器，則所述場景分類器依據所述至少一幀圖片對應的場景類別確定所述待檢測視頻的場景類別，包括：

利用所述場景分類器對所述多幀圖片進行場景分類，確定出每幀圖片對應的場景類別；

依據每幀圖片對應的場景類別對所述多幀圖片分組，統計每組圖片包含的圖片幀數；

將圖片幀數最多的一組圖片對應的場景類別確定為所述待檢測視頻的場景類別。

3.根據權利要求1所述的方法，其中，對所述多幀圖片中的移動物體采用預置目標跟蹤算法進行跟蹤，對跟蹤到的移動物體標注目標框，包括：

定義所述待檢測視頻中已標注目標框的任一幀圖片為跟蹤圖片，所述多幀圖片中與跟蹤圖片相鄰的后幀圖片為被跟蹤圖片；

采用預置目標跟蹤算法在被跟蹤圖片中跟蹤與所述跟蹤圖片中已由移動物體檢測算法標注目標框的移動物體對應的移動物體，在被跟蹤圖片中對跟蹤到的移動物體標注目標框和對應的跟蹤置信度；

以當前新標注后的被跟蹤圖片作為新的跟蹤圖片，對新的跟蹤圖片相鄰的后幀被跟蹤圖片中的移動物體采用預置目標跟蹤算法跟蹤標注，直到所述多幀圖片均標注完成。

4.根據權利要求3所述的方法，其中，采用預設算法在所述多幀圖片中移除所述移動物體上重疊的目標框，包括：

確定所述多幀圖片的移動物體上重疊目標框中每個目標框的置信度；

采用預設算法從所述移動物體上的重疊目標框中移除置信度低于預設值的目標框；

其中，若目標框直接由移動物體檢測算法標注，則該目標框的置信度為所述移動物體檢測算法對應的置信度；

若目標框由預置目標跟蹤算法跟蹤標注，則該目標框的置信度為跟蹤圖片中對應的移動物體在標注目標框時采用的移動物體檢測算法對應的置信度、與所述被跟蹤圖片中標注目標框對應的跟蹤置信度的乘積。

5.根據權利要求1-4任一項所述的方法，其中，所述預設算法包括Soft NMS算法。

6.根據權利要求1-4任一項所述的方法，其中，采用預設算法在所述多幀圖片中移除所述移動物體上重疊的目標框之后，還包括：

接收用戶對所述多幀圖片中移動物體的目標框的檢測、調整操作；

對檢測、調整操作后的多幀圖片進行保存。

7.根據權利要求1-4任一項所述的方法，其中，對所述多幀圖片中的移動物體采用預置目標跟蹤算法進行跟蹤之前，還包括：

對所述待檢測視頻中的多幀圖片進行尺寸縮放；

通過所述移動物體檢測算法對不同尺寸下的多幀圖片中的移動物體分別標注目標框；

將標注完目標框的不同尺寸的多幀圖片縮放還原至初始尺寸。