- 专利标题: 一种基于注意力机制改进的轻量化YOLO v4安全防护检测方法
-
申请号: CN202110782188.X申请日: 2021-07-12
-
公开(公告)号: CN113516076B公开(公告)日: 2023-09-01
- 发明人: 王巍 , 云健 , 张建新 , 于洋跃 , 多俊杰 , 刘勇奎
- 申请人: 大连民族大学
- 申请人地址: 辽宁省大连市开发区辽河西路18号
- 专利权人: 大连民族大学
- 当前专利权人: 大连民族大学
- 当前专利权人地址: 辽宁省大连市开发区辽河西路18号
- 代理机构: 大连一通专利代理事务所
- 代理商 秦少林
- 主分类号: G06V40/10
- IPC分类号: G06V40/10 ; G06V20/52 ; G06V10/44 ; G06V10/764 ; G06V10/82 ; G06N3/0464 ; G06N3/048 ; G06N3/08
摘要:
本发明公开了一种基于注意力机制改进的轻量化YOLO v4安全防护检测方法,其内容包括:搜集图像建立数据集,采用YOLO v4网络作为基础模型,将注意力机制融入到YOLO v4模型中,得到改进的YOLO v4安全防护检测模型;利用数据集对改进的YOLO v4安全防护检测模型进行训练得到改进后的YOLO v4目标检测模型;在改进的YOLO v4安全防护检测模型中输入待检测图像,采用改进后的YOLO v4目标检测模型输出对应的目标检测结果,分别定位行人的检测框、安全帽的检测框以及安全反光衣的检测框,并分别计算三个目标的CIoU,得到最终目标检测框;对施工场景安全设备目标进行检测和识别,并根据目标行人、安全帽和安全反光衣检测框的重合情况,判断图像中是否存在不规范穿戴安全设备的行为。
公开/授权文献
- CN113516076A 一种基于注意力机制改进的轻量化YOLO v4安全防护检测方法 公开/授权日:2021-10-19