[發明專利]一種安全帽佩戴檢測的圖像數據集冗余清洗方法

在審

• 申請號: 202111360409.0
• 申請日: 2021-11-17
• 公開（公告）號: CN114067135A
• 公開（公告）日: 2022-02-18
• 發明（設計）人: 陳軼;張文;王茂森;牛少彰;崔浩亮;馮亞輝;王讓定
• 申請（專利權）人: 東南數字經濟發展研究院;浙江磚助智連科技有限責任公司
• 主分類號: G06V10/72
• 分類號: G06V10/72;G06V10/74;G06V10/774;G06V20/52;G06V20/40;G06K9/62
• 代理公司: 北京盛凡佳華專利代理事務所(普通合伙) 11947
• 代理人: 陳文麗

1.一種安全帽佩戴檢測的圖像數據集冗余清洗方法，其特征在于：包括以下步驟：

S1，使用目標檢測算法YOLO V4算法在VOC數據集上進行訓練，以反向傳播算法進行參數調優，選擇最優結果的模型文件進行保存；

S2，利用FFmpeg將攝像頭采集到的視頻解析成圖像；

S3，解析完成的圖像按照順序輸入步驟S1中訓練完成的目標檢測模型，檢測結果僅保留檢測到人的圖像；

S4，判斷前后兩張圖檢測到的人員數量是否一致；

S5，如果步驟S4中判斷結果為不一致，則將兩張圖像均保存；

S6，如果步驟S4中判斷結果為一致，則計算兩張圖像中人員各個人員位置中心點坐標，將兩張圖之間中心點坐標距離最近的兩個中心點進行對應；

S7，計算各個對應中心點之間的距離之和；

S8，判斷計算距離之和是否大于設定閾值；

S9，如果S8中判斷結果為否，則刪除其中一張圖；

S10，如果S8中判斷結果為是，則兩張圖像均保存。

2.根據權利要求1所述的一種安全帽佩戴檢測的圖像數據集冗余清洗方法，其特征在于：所述步驟S3中輸出中僅保留檢測到人的結果數據替換為VOC數據集中標注為person的結果數據。

3.根據權利要求1所述的一種安全帽佩戴檢測的圖像數據集冗余清洗方法，其特征在于：所述步驟S2具體為利用FFmpeg將攝像頭拍攝到的視頻文件按照設定的幀率解析成圖像格式保存。

4.根據權利要求1所述的一種安全帽佩戴檢測的圖像數據集冗余清洗方法，其特征在于：所述步驟S5具體為如果兩張圖像檢測到的人員數量信息之間不一致，則判定兩張圖不相似，不作為冗余對其刪除。

5.根據權利要求1所述的一種安全帽佩戴檢測的圖像數據集冗余清洗方法，其特征在于：所述步驟S6具體為如果兩張圖像檢測到的人員數量信息之間一致，則分別計算檢測結果中各個人員位置中心點坐標，前一張圖像中對應的各個人員中心點坐標分別與后一張圖像中距離最近的人員中心坐標點進行對應。

6.根據權利要求1所述的一種安全帽佩戴檢測的圖像數據集冗余清洗方法，其特征在于：所述步驟S9具體為如果中心坐標點間距離之和小于等于閾值則表示兩張圖之間相似，其中一張圖是冗余并將其刪除。

7.根據權利要求1所述的一種安全帽佩戴檢測的圖像數據集冗余清洗方法，其特征在于：所述步驟S10具體為如果中心坐標點間距離之和大于閾值則表示兩張圖之間不相似，不存在冗余，兩張圖像均保存。