一種基于YOLOv4-Tiny的加油站安全帽佩戴實時檢測方法

基本信息

申請?zhí)?/td> CN202111495511.1 申請日 -
公開（公告）號 CN114140750A 公開（公告）日 2022-03-04
申請公布號 CN114140750A 申請公布日 2022-03-04
分類號 G06V20/52（2022.01）I;G06V10/764（2022.01）I;G06V10/44（2022.01）I;G06V10/82（2022.01）I;G06N3/04（2006.01）I;G06N3/08（2006.01）I 分類 計算；推算；計數;
發(fā)明人 范慶來;周君良;倪勇龍;陳義;錢至遠;王豆;楊杰;王崇 申請（專利權）人 浙江浙石油綜合能源銷售有限公司
代理機構 浙江千克知識產權代理有限公司 代理人 黎雙華
地址 310000浙江省杭州市西湖區(qū)古蕩街道文二路391號西湖國際科技大廈22樓
法律狀態(tài) -

摘要

摘要 本發(fā)明公開一種基于YOLOv4?Tiny的加油站安全帽佩戴實時檢測方法，包括：S1、對加油站原有監(jiān)控視頻進行抽幀處理，提取得到多張監(jiān)控圖片，對監(jiān)控圖片進行預處理，以得到訓練圖片，并選取相應比例訓練圖片進行對抗樣本生成，剩余訓練圖片作為原始樣本；S2、基于對抗樣本、原始樣本生成第一訓練集；S3、采用第一訓練集對YOLOv4?Tiny改進模型進行訓練，以得到相應訓練權重；S4、接入加油站實時監(jiān)控，并進行幀分解，將每一幀的分解圖片實時輸入訓練完成的YOLOv4?Tiny改進模型中，以得到分解圖片中人員安全帽佩戴分類結果，并將分類結果疊加至分解圖片中，以得到分類圖片；S5、將分類圖片實時合成視頻，進行視頻實時輸出。檢測準確度更高、實時性更強，且適配于油站內硬件設備基礎。