[发明专利]一种基于元学习的变电站改扩建违章行为智能化识别方法

权利要求书

1.一种基于元学习的变电站改扩建违章行为智能化识别方法，其特征在于，包括以下步骤：

(1)采集作业人员、设备、危险区域标志物图片，并构造难样本，完成场景的标注，形成小样本数据集；

(2)在ImageNet数据集上利用元学习方法预训练YOLOv5模型；

(3)在步骤1采集的小样本数据集上精调得到最终的YOLOv5模型；

(4)将训练好的YOLOv5模型部署到移动端，完成对作业人员、施工机具、送变电设备检测物的识别；

(5)根据施工作业要求自适应设置虚拟电子围栏，进行地面作业区域越界检测和高空作业区域越界检测，基于所设置的虚拟围栏进行人员机具的越界违章行为智能化识别并告警；

所述步骤(2)包括以下步骤：

(21)在开源数据集ImageNet上反复随机抽取训练任务，形成一个由若干个任务组成的任务池，作为MAML的训练集；每个任务都有n个类别的样本，每个类别的样本数量相同，任务中包含支持集与查询集，支持集与查询集均包含样本数据与标签；

(22)随机初始化YOLOv5模型的参数；

(23)随机采样k次任务进行训练；

(24)随机采样n个任务，形成一个batch；

(25)利用batch中的每一个任务中的支持集，分别对模型的参数进行更新，得到n个模型；

(26)利用batch中的每一个任务中的查询集，基于在各自支持集上训练得到的模型计算损失，计算一个batch的损失总和，在原模型上进行随机梯度下降，返回(24)；

(27)K次训练结束，得到元模型YOLOv5_meta；

所述步骤(3)包括以下步骤：

(31)将采集到的变电站改扩建工作场景数据分为支持集与查询集；

(32)导入训练好的YOLOv5_meta的初始化参数；

(33)利用支持集，对模型的参数进行更新；

(34)得到最终的模型YOLOv5_fine-tune，利用查询集测试模型性能。

2.根据权利要求1所述基于元学习的变电站改扩建违章行为智能化识别方法，其特征在于，所述步骤(1)包括以下步骤：

(11)采集的数据集来自实际变电站改扩建工作场景，针对不同远近距离下目标尺寸差异大的问题，构造数据集时考虑不同尺度下的人员、机具、送变电设备和标志物图片，使近、中、远距离图片数量相近；

(12)采集到的图片数据集用图形图像注释工具LabelImg打标签，将图片中的人员、机具、送变电设备、危险区域标志物标注出来，获得目标物体的坐标与宽高信息，制作数据集的标签，生成xml文件。

3.根据权利要求1所述基于元学习的变电站改扩建违章行为智能化识别方法，其特征在于，所述步骤(4)包括以下步骤：

(41)将PyTorch框架下训练好的YOLOv5模型转为ONNX模型，再将ONNX模型转为NCNN模型即移动端的YOLOv5模型；

(42)用C++的代码重构违章检测代码；

(43)对移动端拍摄的视频进行处理，将每帧图片送入移动端的YOLOv5模型；

(44)将YOLOv5模型的输出送入解码函数中进行解码，得到检测物的类别信息、置信度信息和坐标宽高信息。

4.根据权利要求1所述基于元学习的变电站改扩建违章行为智能化识别方法，其特征在于，步骤(5)所述的地面作业区域越界检测实现过程如下：

通过YOLOv5模型获取人员、机具和作业区域边界标志物检测框的坐标宽高信息；通过判断各边界标志物的几何约束，在场景中划分出多个作业区域，将属于同一区域的边界标志物所组成的多边形标定为一个作业区域；分别计算出距人员与机具最近的作业区域，根据射线法判断人员和机具是否在作业区域内。

5.根据权利要求1所述基于元学习的变电站改扩建违章行为智能化识别方法，其特征在于，步骤(5)所述的高空作业区域越界检测实现过程如下：

获取变电站高空可作业高度h，测量人员标志物的宽度w_标，测量人员的安全帽直径w_帽，计算可作业高度h与标志物的宽度w_标的比例k_高＝H/w_标，计算人员标志物的宽度w_标与人员的安全帽直径w_帽的比例k_人＝w_标/w_帽；

通过YOLOv5模型获取作业人员佩戴的安全帽检测框的坐标宽高信息，根据人员标志物的宽度与安全帽的直径比例k_人计算该作业人员所属的人员标志物，根据可作业高度与标志物的宽度的比例k_高计算该作业人员的最大可至高度；根据作业人员与所属的人员标志物检测框的坐标计算图片中作业人员距离地面的高度，并与最大可至高度进行比较，判断作业人员是否越界。