基于YOLOv5的工地行为安全检测识别方法及系统技术方案

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

【技术实现步骤摘要】
基于YOLOv5的工地行为安全检测识别方法及系统
本专利技术涉及建筑工程领域，尤其涉及基于YOLOv5的工地行为安全检测识别方法及系统。
技术介绍
建筑施工作业过程中，存在着较多安全隐患，安全事故的发生率居高不下。在长期的实践论证中，在开展施工作业之前，对建筑工人的行为能力和安全设施的佩戴进行检查，能够有效的减少事故发生的几率。因此，在日常施工作业工程中，监督工人的安全帽等安全设施是否佩戴显得尤为重要。但是目前大部分工地上，采用的都是人为监测，很依赖于现场经验丰富的管理人员，需要他们实时的进行观察和检查，又费时又费力，存在着自动化水平较低，工作量较大，检查项目有限的情况，极容易出现漏检等情况，造成安全隐患。因此出现了工地监管系统，实现智能化地监督工人施工。传统工地监管系统大多由专门的工程管理干系实现，但是此种的工地监管系统需要消耗大量人力，并且由于个人疏忽等其他难以回避的因素，很难做到安全规范的广泛落实，具有可靠性较低、反应时延较长、成本较大以及管理不便等显著缺陷。因此出现了基于目标检测技术的自动化识别的工地监管系统，可以摆脱对人力的依赖，提供可靠、可控、低成本的工地安全保障，目标检测即对待检测对象的精准定位与准确识别，确定目标所在的位置和具体类别，当前对于目标检测方法的研究已经相当成熟了。不过对于施工现场，一些环境比较复杂的工矿这些特殊场景中对于目标检测的实时性、精确度都有着相当高的要求，对于这些外在的干扰特征信息，机器很多情况下无法做出准确判断，目标检测物的实时性、精确度较低，且通用性不强。>
技术实现思路
有鉴于此，本专利技术的目的在于提出基于YOLOv5的工地行为安全检测识别方法及系统，以解决上述
技术介绍
中提及的全部问题或之一。基于上述目的，本专利技术提供了基于YOLOv5的工地行为安全检测识别方法，包括以下步骤：获取待识别的工地现场图像信息；采用预训练好的检测模型对所述待识别的工地现场图像信息进行识别，对安全帽的佩戴情况进行检测，其中，检测模型为深度学习模型，用于训练检测模型的训练集包括标注安全帽佩戴位置的图像信息；当识别到工人未佩戴安全帽或者佩戴位置不规范时，发出警示。可选的，所述检测模型为Yolov5模型。可选的，获取待识别的工地现场图像信息之前，还包括：搭建所述检测模型，检测模型具有Yolov5模型结构的pytorch框架；采集工地现场中不同环境下工人的工地现场图像信息；响应于所述工地现场图像信息中包括有待筛选的模糊图像的区域，将所述模糊图像的区域确定为可疑候选区域；将所述可疑候选区域切割成尺寸一致的多个区块，根据各个所述区块中安全帽颜色像素的占比，确定可疑候选区域的分割区块，其中，所述安全帽颜色像素为在安全帽目标颜色色域内的像素，通过判断像素的RGB值确定是否为安全帽颜色像素；根据所述分割区块，确定安全帽的佩戴位置，并对安全帽的佩戴位置进行标注，得到均包括标注安全帽佩戴位置的图像信息的训练集和测试集；采用所述训练集对检测模型进行训练，并根据训练结果对检测模型进行参数微调，得到预训练好的检测模型。可选的，响应于所述工地现场图像信息中包括有待筛选的模糊图像的区域，将所述模糊图像的区域确定为可疑候选区域之前，还包括：响应于任何一个区域内安全帽颜色像素的占比不高于预设阈值，确定该工地现场图像信息中不包括有安全帽图像信息，剔除该工地现场图像信息。可选的，响应于所述工地现场图像信息中包括有待筛选的模糊图像的区域，将所述模糊图像的区域确定为可疑候选区域，具体包括：将所述工地现场图像信息划分为多个区域；获取每个所述区域内安全帽颜色的占比，所述占比的计算方法为公式(1)，所述公式(1)为：响应于任意一个区域内安全帽颜色的占比低于预设阈值，将安全帽颜色的占比低于预设阈值的区域确定为待筛选的模糊图像的区域；响应于工地现场图像信息中包括有待筛选的模糊图像的区域，将安全帽颜色的占比低于预设阈值的区域确定为可疑候选区域。可选的，将所述可疑候选区域切割成尺寸一致的多个区块，具体为：将可疑候选区域按等高或者等宽的原则切割成尺寸一致的多个区块。可选的，将所述可疑候选区域切割成尺寸一致的多个区块，根据各个所述区块中安全帽颜色像素的占比，确定可疑候选区域的分割区块，具体包括：确定基础区块；从所述基础区块开始，获取基础区块后的每个区块中安全帽颜色像素的占比；获取占比最小的区块，将占比最小的区块确定为可疑候选区域的分割区块。可选的，所述训练集和所述测试集的比例为7：3。可选的，采用所述训练集对检测模型进行训练，并根据训练结果对检测模型进行参数微调，得到预训练好的检测模型中，所述参数包括P、R和mAP，P、R和mAP分别代表识别精度、召回率和综合准确率。进一步的，本专利技术还提供了基于YOLOv5的工地行为安全检测识别系统，包括：现场信息获取模块，用于获取待识别的工地现场图像信息；安全帽位置识别模块，用于采用预训练好的检测模型对所述待识别的工地现场图像信息进行识别，对安全帽的佩戴情况进行检测，其中，检测模型为深度学习模型，用于训练检测模型的训练集包括标注安全帽佩戴位置的图像信息；安全警示模块，用于当识别到工人未佩戴安全帽或者佩戴位置不规范时，发出警示。从上面所述可以看出，本专利技术提供的基于YOLOv5的工地行为安全检测识别方法，具备以下有益效果：该方法和系统中通过采集工地现场中不同环境下工人的工地现场图像，并对工地现场图像进行筛选，制作专有的训练集，专有的训练集具有多样性，使得利用专有的训练集训练得到检测模型能够准确快速的安全帽和安全帽的位置信息，目标检测模型为于YOLOv5模型，使得到的目标检测模型不仅具有高速数据处理能力，能够提高识别的效率，还能快速移植于其他的移动设备，通用性强。在对工地现场图像进行筛选的过程中，通过对工地现场图像的可疑候选区域进行区块划分，并根据各区块包括的安全帽颜色的占比来确定作为分界线的目标区块并进行图像切割，得到安全帽的佩戴位置，具有较高的识别准确率。该方法和系统整合机器视觉技术、智慧工地技术、神经网络技术的一体化工程安全监管应用成果，实现对工地现场工人安全帽佩戴情况和佩戴是否规范的实时监测与告警，以此加强工程管理规范性、提升工程整体效率、减少成本消耗、降低意外事故与人员伤亡发生概率、并推进智慧城市与智慧工地建设等。附图说明为了更清楚地说明本专利技术实施例或现有技术中的技术方案，下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本专利技术的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。图1为本专利技术方法的流程示意图一；图2为本专利技术方法的流程示意图二；图3为本专利技术方法的流程示意图三；图4为本专利技术方法中S300步骤具体的流本文档来自技高网...

【技术保护点】
1.基于YOLOv5的工地行为安全检测识别方法，其特征在于：包括以下步骤：/n获取待识别的工地现场图像信息；/n采用预训练好的检测模型对所述待识别的工地现场图像信息进行识别，对安全帽的佩戴情况进行检测，其中，检测模型为深度学习模型，用于训练检测模型的训练集包括标注安全帽佩戴位置的图像信息；/n当识别到工人未佩戴安全帽或者佩戴位置不规范时，发出警示。/n

【技术特征摘要】
1.基于YOLOv5的工地行为安全检测识别方法，其特征在于：包括以下步骤：
获取待识别的工地现场图像信息；
采用预训练好的检测模型对所述待识别的工地现场图像信息进行识别，对安全帽的佩戴情况进行检测，其中，检测模型为深度学习模型，用于训练检测模型的训练集包括标注安全帽佩戴位置的图像信息；
当识别到工人未佩戴安全帽或者佩戴位置不规范时，发出警示。


2.根据权利要求1所述的基于YOLOv5的工地行为安全检测识别方法，其特征在于：所述检测模型为Yolov5模型。


3.根据权利要求2所述的基于YOLOv5的工地行为安全检测识别方法，其特征在于：获取待识别的工地现场图像信息之前，还包括：
搭建所述检测模型，检测模型具有Yolov5模型结构的pytorch框架；
采集工地现场中不同环境下工人的工地现场图像信息；
响应于所述工地现场图像信息中包括有待筛选的模糊图像的区域，将所述模糊图像的区域确定为可疑候选区域；
将所述可疑候选区域切割成尺寸一致的多个区块，根据各个所述区块中安全帽颜色像素的占比，确定可疑候选区域的分割区块，其中，所述安全帽颜色像素为在安全帽目标颜色色域内的像素，通过判断像素的RGB值确定是否为安全帽颜色像素；
根据所述分割区块，确定安全帽的佩戴位置，并对安全帽的佩戴位置进行标注，得到均包括标注安全帽佩戴位置的图像信息的训练集和测试集；
采用所述训练集对检测模型进行训练，并根据训练结果对检测模型进行参数微调，得到预训练好的检测模型。


4.根据权利要求3所述的基于YOLOv5的工地行为安全检测识别方法，其特征在于：响应于所述工地现场图像信息中包括有待筛选的模糊图像的区域，将所述模糊图像的区域确定为可疑候选区域之前，还包括：
响应于任何一个区域内安全帽颜色像素的占比不高于预设阈值，确定该工地现场图像信息中不包括有安全帽图像信息，剔除该工地现场图像信息。


5.根据权利要求4所述的基于YOLOv5的工地行为安全检测识别方法，其特征在于：响应于所述工地现场图像信息中包括有待筛选的模糊图像的区域，将所述模糊图像的区域确定为可疑候选区域，具...

【专利技术属性】
技术研发人员：鄢必超，
申请(专利权)人：深圳市安比智慧科技有限公司，
类型：发明
国别省市：广东;44