一种基于DSSD算法的安全帽佩戴检查方法、装置与系统

技术领域

本发明涉及图像识别技术领域，特别是一种基于DSSD算法的安全帽佩戴检查方法、装置与系统。

背景技术

在电力施工现场，安全帽在施工人员安全防护中起到重要的作用，是头部防护的必备物件。然而，由于施工人员一时疏忽或安全意识不足，经常在施工现场存在不佩戴安全帽的现象，使得产生了一定的安全隐患。因此，要求管理人员如何对进行施工中的人员安全帽佩戴情况进行检查。但是由于电力施工的环境复杂，人工检查效率低且容易出现遗漏，安全帽检测的准确性较低。随着基于深度学习的图像识别等人工智能技术的飞速发展，采用智能图像识别技术进行安全帽检查，具有相对高效和高准确率，使得其替代人工检查成为可能。

名词解释

DSSD（Deconvolutional Single Shot Detector）算法，借鉴了FPN的思路，将ResNet101 作为特征提取网络，是对SSD算法的优化改进。

发明内容

本发明的目的是提供一种基于DSSD算法的安全帽佩戴检查方法、装置与系统，能够根据采集到的图像，对电力施工环境中人员佩戴安全帽的情况进行检查，提升安全检查效率，且结果较可靠。本发明采用的技术方案如下。

一方面，本发明提供一种安全帽佩戴检查方法，包括：

获取从施工环境中采集的图像；

对获取到的图像进行预处理；

将预处理得到的图像输入预先训练的安全帽佩戴检查模型；

根据所述安全帽佩戴检查模型的输出，得到图像中人员的安全帽佩戴结果信息；

所述预先训练的安全帽佩戴检查模型采用DSSD算法模型，训练样本包括公共数据集以及自定义样本集，所述自定义样本集包括已标注的人员佩戴安全帽的图像样本以及人员未佩戴安全帽的图像样本。

可选的，所述公共数据集采用VOC2007的trainval数据集，自定义样本集包括归一化处理后的数量比例为1： 1的人员佩戴安全帽的图像样本和人员未佩戴安全帽的图像样本，图像样本的分辨率为512x512；

所述对获取到的图像进行预处理包括对图像进行归一化处理，得到分辨率为512x512的图像。

可选的，DSSD算法模型包括基础网络、反卷积网络和预测网络；

所述安全帽佩戴检查模型的训练方法包括：

采用ResNet101作为基础网络；

从公共数据集中选择训练样本集及测试样本集，对基础网络进行训练；

冻结ResNet101网络在初步训练后的网络参数，加载反卷积网络；

利用从公共数据集中选择的训练样本集及测试样本集，对反卷积网络进行训练；

解冻基础网络参数，加载预测网络；

利用从公共数据集中选择的训练样本集及测试样本集，对当前全部网络进行训练；

从自定义样本集中选择训练样本集及测试样本集，对经公共数据集训练后的全部网络再次进行训练，得到所述安全帽佩戴检查模型。

可选的，对基础网络进行训练的过程中共迭代7万次，前4万次采用1e-3的学习率，中间2万次采用1e-3的学习率，最后1万次采用1e-5的学习率。

可选的，对反卷积网络进行训练的过程中共迭代3万次，前2万次和后一万次的学习率分别为1e-3、1e-4。

可选的，利用公共数据集对完整网络进行训练的过程中共迭代4万次，前2万次和后两万次的学习率分别设置为1e-3、1e-4。

可选的，利用自定义样本集进行训练的过程中共迭代4万次，前2万次和后两万次的学习率分别设置为1e-3、1e-4。

第二方面，本发明提供一种安全帽佩戴检查装置，包括：

图像获取模块，被配置用于获取从施工环境中采集的图像；

图像预处理模块，被配置用于对获取到的图像进行预处理；

图像识别模块，被配置用于将预处理得到的图像输入预先训练的安全帽佩戴检查模型，检查图像中的人员是否佩戴安全帽；

以及，检查结果输出模块，被配置用于根据所述安全帽佩戴检查模型的输出，得到图像中人员的安全帽佩戴检查结果信息；

其中所述预先训练的安全帽佩戴检查模型采用DSSD算法模型，其训练样本包括公共数据集以及自定义样本集，所述自定义样本集包括已标注的人员佩戴安全帽的图像样本以及人员未佩戴安全帽的图像样本。

可选的，所述检查结果输出模块包括：

实时展示单元，用于实时展示人员佩戴及未佩戴安全帽的图像；

实时告警单元，用于实时输出人员未佩戴安全帽的告警信息；

信息统计单元，用于统计人员佩戴以及未佩戴安全帽的信息；

以及数据存储单元，用于存储人员佩戴以及未佩戴安全帽的图像及信息统计单元得到的统计信息。

第三方面，本发明提供一种安全帽佩戴检查系统，包括在施工现场多点设置的图像采集设备和安全帽佩戴检查设备；

图像采集设备采集安装点对应位置处的施工人员图像，传输至所述安全帽佩戴检查设备；

安全帽佩戴检查设备将获取到的施工人员图像作为待检查图像，执行第一方面所述的安全帽佩戴检查方法，输出安全帽佩戴检查结果信息。

有益效果

本发明通过采用DSSD算法进行安全帽检查测试，避免了在施工环境中安置检查专员的人力投入。而且实验证明，在相同的软硬件条件下，采用DSSD算法的检测识别精度比采用SSD算法提高约2.2%，比采用Faster R-CNN算法提高约0.9%，而检测速度与SSD算法相当，比Faster R-CNN算法提高了约9倍。因此，本发明在电力施工人员安全帽检测领域具有较高的技术应用价值。

附图说明

图1所示为DSSD算法模型整体网络构架图；

图2所示为DSSD算法模型中反卷积网络结构图；

图3所示为DSSD算法模型中预测网络结构图；

图4所示为安全帽佩戴检查系统的一种实施例示意图。

具体实施方式

以下结合附图和具体实施例进一步描述。

实施例1

本实施例介绍一种安全帽佩戴检查方法，参考图4，方法包括：

获取从施工环境中采集的图像；

对获取到的图像进行预处理；

将预处理得到的图像输入预先训练的安全帽佩戴检查模型；

根据所述安全帽佩戴检查模型的输出，得到图像中人员的安全帽佩戴结果信息；

所述预先训练的安全帽佩戴检查模型采用DSSD算法模型，训练样本包括公共数据集以及自定义样本集，所述自定义样本集包括已标注的人员佩戴安全帽的图像样本以及人员未佩戴安全帽的图像样本。

本实施例方法的具体实现涉及以下内容。

一、安全帽佩戴检查模型的构建

编程实现DSSD算法网络：参考图1，将SSD的基础网络由VGG网络构架换成Residual-101网络构架。对最深一层特征图层连续做5次反卷积操作，每次反卷积操作后的特征图层尺寸都与不同深度的特征图像尺度对应相同，再采用跳步连接反卷积特征图引入，使之与相同尺寸的特征图层采用图2的网络架构，融合在一起，得到的新的特征图层，形成多尺度检测框架，用来做预测。采用3个卷积层与矩阵点积运算层构成预测网络，实现目标预测和逻辑回归，如图3所示。本实施例将默认的预测锚框长宽比例为1、1.6、2和3。

二、安全帽佩戴检查模型的训练

2.1构建训练数据集

在对DSSD模型进行训练时，可首先采用公共数据集作为训练数据集，公共数据集可采用VOC2007的trainval数据集。

DSSD算法模型的迁移学习利用自定义样本集实现。可通过拍照和视频剪辑的方式，收集在电力施工现场佩戴安全帽和未佩戴的安全帽的人像。对图像进行归一化统一分辨率为512x512，并对人像进行分类标注，制作标签为两类，分别带安全帽人像与不带安全帽人像。按照标签比例1:1，构建自定义样本集。

2.2 模型训练过程

首先采用ResNet101作为基础网络，采用VOC2007的trainval数据集和测试集分别作为模型训练数据集和测试集。训练时一共迭代7万次，前4万次采用1e-3的学习率，中间2万次采用1e-3的学习率，最后1万次采用1e-5的学习率。

然后，冻结基础网络的网络参数，增加反卷积网络(不添加预测网络)，仍采用VOC2007的trainval数据集和测试集作为模型训练数据集和测试集，在基础网络参数冻结的条件下将只训练反卷积网络。训练过程设置学习率为1e-3、1e-4分别进行2万次和1万次的迭代训练。

接着，解冻基础网络的网络参数，并添加预测网络，设置学习率为1e-3、1e-4再分别都进行2万次的迭代训练。完成在VOC2007数据集上的模型训练。

最后，迁移学习。基于自定义数据集，按照95：5比例构建训练集和测试集，在VOC2007训练好的模型基础上，设置学习率为1e-3、1e-4再分别都进行2万次的迭代训练，最终得到基于DSSD算法安全帽佩戴检查算法模型文件。

三、安全帽佩戴检查应用

在进行施工环境安全帽佩戴检查时，可通过图像采集设备采集施工环境中的人员图像，然后对图像进行预处理，如去噪、归一化等常规处理。再将预处理后的图像输入至训练好的DSSD算法模型中，即可得到安全帽佩戴检查结果。

本实施例经训练得到的DSSD算法模型为端到端模型，输入图像后可输出带安全帽和不带安全帽的人员统计信息，以及图像上的人员框图及标签信息。

可将本发明的安全帽佩戴检查方法集成至电力施工在线监控系统中，丰富电力施工在线监控系统的监控功能，保障施工现场安全。

实施例2

本实施例介绍一种安全帽佩戴检查装置，包括：

图像获取模块，被配置用于获取从施工环境中采集的图像；

图像预处理模块，被配置用于对获取到的图像进行预处理；

图像识别模块，被配置用于将预处理得到的图像输入预先训练的安全帽佩戴检查模型，检查图像中的人员是否佩戴安全帽；

以及，检查结果输出模块，被配置用于根据所述安全帽佩戴检查模型的输出，得到图像中人员的安全帽佩戴检查结果信息；

其中所述预先训练的安全帽佩戴检查模型采用DSSD算法模型，其训练样本包括公共数据集以及自定义样本集，所述自定义样本集包括已标注的人员佩戴安全帽的图像样本以及人员未佩戴安全帽的图像样本。

以上各模块的具体功能实现可参考实施例1方法中的相关介绍。

进一步的，本实施例检查结果输出模块包括：

实时展示单元，用于实时展示人员佩戴及未佩戴安全帽的图像；

实时告警单元，用于实时输出人员未佩戴安全帽的告警信息；

信息统计单元，用于统计人员佩戴以及未佩戴安全帽的信息；

以及数据存储单元，用于存储人员佩戴以及未佩戴安全帽的图像及信息统计单元得到的统计信息。由此可实现施工现场更多方位的人机交互，进一步提升施工安全性，并方便信息的后期追溯。

实施例3

参考图4，本实施例介绍一种安全帽佩戴检查系统，包括在施工现场多点设置的图像采集设备和安全帽佩戴检查设备；

图像采集设备采集安装点对应位置处的施工人员图像，传输至所述安全帽佩戴检查设备；

安全帽佩戴检查设备将获取到的施工人员图像作为待检查图像，执行第一方面所述的安全帽佩戴检查方法，输出安全帽佩戴检查结果信息。

输出的检查结果信息可以是图像信息、报警信息等，方便监控人员的监控，提升施工安全性。

本领域内的技术人员应明白，本申请的实施例可提供为方法、系统、或计算机程序产品。因此，本申请可采用完全硬件实施例、完全软件实施例、或结合软件和硬件方面的实施例的形式。而且，本申请可采用在一个或多个其中包含有计算机可用程序代码的计算机可用存储介质（包括但不限于磁盘存储器、CD-ROM、光学存储器等）上实施的计算机程序产品的形式。

本申请是参照根据本申请实施例的方法、设备（系统）、和计算机程序产品的流程图和／或方框图来描述的。应理解可由计算机程序指令实现流程图和／或方框图中的每一流程和／或方框、以及流程图和／或方框图中的流程和／或方框的结合。可提供这些计算机程序指令到通用计算机、专用计算机、嵌入式处理机或其他可编程数据处理设备的处理器以产生一个机器，使得通过计算机或其他可编程数据处理设备的处理器执行的指令产生用于实现在流程图一个流程或多个流程和／或方框图一个方框或多个方框中指定的功能的装置。

这些计算机程序指令也可存储在能引导计算机或其他可编程数据处理设备以特定方式工作的计算机可读存储器中，使得存储在该计算机可读存储器中的指令产生包括指令装置的制造品，该指令装置实现在流程图一个流程或多个流程和／或方框图一个方框或多个方框中指定的功能。

这些计算机程序指令也可装载到计算机或其他可编程数据处理设备上，使得在计算机或其他可编程设备上执行一系列操作步骤以产生计算机实现的处理，从而在计算机或其他可编程设备上执行的指令提供用于实现在流程图一个流程或多个流程和／或方框图一个方框或多个方框中指定的功能的步骤。

以上结合附图对本发明的实施例进行了描述，但是本发明并不局限于上述的具体实施方式，上述的具体实施方式仅仅是示意性的，而不是限制性的，本领域的普通技术人员在本发明的启示下，在不脱离本发明宗旨和权利要求所保护的范围情况下，还可做出很多形式，这些均属于本发明的保护之内。