一种基于注意力机制改进的轻量化YOLO v4安全防护检测方法
摘要:
本发明公开了一种基于注意力机制改进的轻量化YOLO v4安全防护检测方法,其内容包括:搜集图像建立数据集,采用YOLO v4网络作为基础模型,将注意力机制融入到YOLO v4模型中,得到改进的YOLO v4安全防护检测模型;利用数据集对改进的YOLO v4安全防护检测模型进行训练得到改进后的YOLO v4目标检测模型;在改进的YOLO v4安全防护检测模型中输入待检测图像,采用改进后的YOLO v4目标检测模型输出对应的目标检测结果,分别定位行人的检测框、安全帽的检测框以及安全反光衣的检测框,并分别计算三个目标的CIoU,得到最终目标检测框;对施工场景安全设备目标进行检测和识别,并根据目标行人、安全帽和安全反光衣检测框的重合情况,判断图像中是否存在不规范穿戴安全设备的行为。
0/0