一种基于YOLOv4-Tiny的加油站安全帽佩戴实时检测方法
基本信息
申请号 | 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 | 分类 | 计算;推算;计数; |
发明人 | 范庆来;周君良;倪勇龙;陈义;钱至远;王豆;杨杰;王崇 | 申请(专利权)人 | 浙江浙石油综合能源销售有限公司 |
代理机构 | 浙江千克知识产权代理有限公司 | 代理人 | 黎双华 |
地址 | 310000浙江省杭州市西湖区古荡街道文二路391号西湖国际科技大厦22楼 | ||
法律状态 | - |
摘要
摘要 | 本发明公开一种基于YOLOv4‑Tiny的加油站安全帽佩戴实时检测方法,包括:S1、对加油站原有监控视频进行抽帧处理,提取得到多张监控图片,对监控图片进行预处理,以得到训练图片,并选取相应比例训练图片进行对抗样本生成,剩余训练图片作为原始样本;S2、基于对抗样本、原始样本生成第一训练集;S3、采用第一训练集对YOLOv4‑Tiny改进模型进行训练,以得到相应训练权重;S4、接入加油站实时监控,并进行帧分解,将每一帧的分解图片实时输入训练完成的YOLOv4‑Tiny改进模型中,以得到分解图片中人员安全帽佩戴分类结果,并将分类结果叠加至分解图片中,以得到分类图片;S5、将分类图片实时合成视频,进行视频实时输出。检测准确度更高、实时性更强,且适配于油站内硬件设备基础。 |
