[发明专利]基于YOLOv5的工地行为安全检测识别方法及系统

说明书

本发明提供了基于YOLOv5的工地行为安全检测识别方法及系统，涉及建筑工程领域，包括以下步骤：获取待识别的工地现场图像信息；采用预训练好的检测模型对所述待识别的工地现场图像信息进行识别，对安全帽的佩戴情况进行检测，其中，检测模型为深度学习模型，用于训练检测模型的训练集包括标注安全帽佩戴位置的图像信息；当识别到工人未佩戴安全帽或者佩戴位置不规范时，发出警示，本发明整合机器视觉技术、智慧工地技术、神经网络技术的一体化工程安全监管应用成果，实现对工地现场工人安全帽佩戴情况的实时监测与告警，以此加强工程管理规范性、提升工程整体效率、减少成本消耗、降低意外事故与人员伤亡发生概率、并推进智慧城市与智慧工地建设等。

技术领域

本发明涉及建筑工程领域，尤其涉及基于YOLOv5的工地行为安全检测识别方法及系统。

背景技术

建筑施工作业过程中，存在着较多安全隐患，安全事故的发生率居高不下。在长期的实践论证中，在开展施工作业之前，对建筑工人的行为能力和安全设施的佩戴进行检查，能够有效的减少事故发生的几率。因此，在日常施工作业工程中，监督工人的安全帽等安全设施是否佩戴显得尤为重要。但是目前大部分工地上，采用的都是人为监测，很依赖于现场经验丰富的管理人员，需要他们实时的进行观察和检查，又费时又费力，存在着自动化水平较低，工作量较大，检查项目有限的情况，极容易出现漏检等情况，造成安全隐患。因此出现了工地监管系统，实现智能化地监督工人施工。

传统工地监管系统大多由专门的工程管理干系实现，但是此种的工地监管系统需要消耗大量人力，并且由于个人疏忽等其他难以回避的因素，很难做到安全规范的广泛落实，具有可靠性较低、反应时延较长、成本较大以及管理不便等显著缺陷。因此出现了基于目标检测技术的自动化识别的工地监管系统，可以摆脱对人力的依赖，提供可靠、可控、低成本的工地安全保障，目标检测即对待检测对象的精准定位与准确识别，确定目标所在的位置和具体类别，当前对于目标检测方法的研究已经相当成熟了。不过对于施工现场，一些环境比较复杂的工矿这些特殊场景中对于目标检测的实时性、精确度都有着相当高的要求，对于这些外在的干扰特征信息，机器很多情况下无法做出准确判断，目标检测物的实时性、精确度较低，且通用性不强。

发明内容

有鉴于此，本发明的目的在于提出基于YOLOv5的工地行为安全检测识别方法及系统，以解决上述背景技术中提及的全部问题或之一。

基于上述目的，本发明提供了基于YOLOv5的工地行为安全检测识别方法，包括以下步骤：

获取待识别的工地现场图像信息；

采用预训练好的检测模型对所述待识别的工地现场图像信息进行识别，对安全帽的佩戴情况进行检测，其中，检测模型为深度学习模型，用于训练检测模型的训练集包括标注安全帽佩戴位置的图像信息；

当识别到工人未佩戴安全帽或者佩戴位置不规范时，发出警示。

可选的，所述检测模型为Yolov5模型。

可选的，获取待识别的工地现场图像信息之前，还包括：

搭建所述检测模型，检测模型具有Yolov5模型结构的pytorch框架；

采集工地现场中不同环境下工人的工地现场图像信息；

响应于所述工地现场图像信息中包括有待筛选的模糊图像的区域，将所述模糊图像的区域确定为可疑候选区域；

将所述可疑候选区域切割成尺寸一致的多个区块，根据各个所述区块中安全帽颜色像素的占比，确定可疑候选区域的分割区块，其中，所述安全帽颜色像素为在安全帽目标颜色色域内的像素，通过判断像素的RGB值确定是否为安全帽颜色像素；

根据所述分割区块，确定安全帽的佩戴位置，并对安全帽的佩戴位置进行标注，得到均包括标注安全帽佩戴位置的图像信息的训练集和测试集；