[發明專利]一種基于中心定位的人頭檢測方法

在審

• 申請號: 202110865599.5
• 申請日: 2021-07-29
• 公開（公告）號: CN113902911A
• 公開（公告）日: 2022-01-07
• 發明（設計）人: 劉栓
• 申請（專利權）人: 四川天翼網絡服務有限公司
• 主分類號: G06V10/26
• 分類號: G06V10/26;G06V10/774;G06V40/10;G06K9/62
• 代理公司: 成都金英專利代理事務所(普通合伙) 51218
• 代理人: 袁英

本發明公開一種基于中心定位的人頭檢測方法，包括以下步驟：S1：選取指定視頻素材并對其進行抽幀處理，將抽幀之后獲得的圖像存入素材庫，選擇目標圖像；S2：利用圖像處理技術對目標圖像進行特征提取和實例分割；S3：建立人頭中心定位模型，基于人頭中心點特征對人頭中心定位模型進行訓練；S4：將S2中處理后的目標圖像輸入訓練后的人頭中心定位模型中，利用人頭中心定位模型處理目標圖像；S5：獲得最終檢測結果。通過本方案依靠點標注和目標函數的設計進行回歸，學習目標中心點的預測，并通過利用基于分水嶺分割方法進行實例分割，提高對相互遮擋人頭的檢測性能。

技術領域

本發明涉及圖像處理領域，尤其涉及一種基于中心定位的人頭檢測方法。

背景技術

基于視頻分析的人流量統計方法可以實現人流量的自動統計，解決人工統計帶來的各種問題。目前，基于視頻分析的流量統計方法主要有三類：

一是基于特征點跟蹤的方法，該方法首先跟蹤一些運動的特征點，然后對特征點的軌跡進行聚類分析，從而得到人流量信息；基于特征點跟蹤的方法需要跟蹤一些運動的特征點，然后對特征點的軌跡進行聚類分析，從而得到人流量信息，該方法的缺點是特征點本身難以穩定地跟蹤，計數精度較差。

二是基于人體分割和跟蹤的方法，該方法首先需要提取出運動目標塊，然后對運動目標塊進行分割得到單個人體目標，最后跟蹤各個人體目標實現人流量的統計；基于人體分割和跟蹤的方法首先需要提取處運動目標塊，然后對運動目標塊進行分割得到單個人體目標，最后跟蹤得到各個人體的軌跡，從而實現人流量的統計。該方法的缺點是當人體存在遮擋時，人體分割的準確性難以得到保證，影響統計精度。

三是基于人頭或頭肩檢測和跟蹤的方法，該方法在視頻中檢測人頭或頭肩，通過對人頭或頭肩的跟蹤進行人流量的統計?；谌祟^數檢測和跟蹤的方法是在視頻中檢測人頭，通過對人頭的跟蹤進行人流量的統計，當攝像機角度適當時，人頭出現遮擋的情況較少，因此基于人頭數檢測的方法較前兩種方法準確性有所提高。

申請號為CN201610689381.8的中國發明申請文件中提出了一種基于Hough變換的人頭檢測方法，并提到：一般采用Hough變換進行人頭的檢測框架是對輸入圖像進行預處理，用canny或者背景差分加sobel算子提取目標邊緣，然后進行Hough橢圓檢測，但這種方法存在一些不足，主要表現在邊緣提取在很多復雜情況下很難得到理想的結果。

現有的人頭檢測方法有兩種，一是基于回歸思想，直接根據圖像回歸出擁擠人群的熱點圖，這種方法存在的缺陷是只能得到場景整體的一個擁堵指數，不能獲取每個人頭所在的具體坐標，而且這種辦法對圖像分辨率很敏感。二是基于目標檢測方法，將人頭作為檢測的目標，這種方法可以得到人頭的數量，也可以獲取人頭所在的具體坐標，但缺點是對小目標人頭、相互遮擋人頭檢測效果很差。并且檢測中的檢測框默認是軸對稱的，大多數比較成功的檢測方法都是要列舉很大數量的候選框，再進一步進行定位和分類。但是這樣的做法往往是多余而又低效率的。

發明內容

本發明的目的在于克服現有技術的不足，提供一種基于中心定位的人頭檢測方法，用于解決利用目標檢測進行人頭檢測必須依賴anchor的問題。

本發明的目的是通過以下技術方案來實現的：

一種基于中心定位的人頭檢測方法，包括以下步驟：

S1：選取指定視頻素材并對其進行抽幀處理，將抽幀之后獲得的圖像存入素材庫，選擇目標圖像；

S2：利用圖像處理技術對目標圖像進行特征提取和實例分割；

S3：建立人頭中心定位模型，基于人頭中心點特征對人頭中心定位模型進行訓練；

S4：將S2中處理后的目標圖像輸入訓練后的人頭中心定位模型中，利用人頭中心定位模型處理目標圖像；

S5：獲得最終檢測結果。

優選的，所述S3包括以下步驟：