一种电力设备自动标注方法和装置与流程

文档序号:22616506发布日期:2020-10-23 19:17阅读:来源:国知局

技术特征:

1.一种电力设备自动标注方法,其特征在于,包括:

步骤1:获得经过设备级标注的电力设备图像集;

步骤2:对图像进行预处理;

步骤3:将电力设备图像集分为训练集和测试集;

步骤4:将训练集中的电力设备图像以及该图像的标签输入到电力设备标注模型进行训练,获得图像中电力设备的预测框和预测设备类型;

步骤5:采用梯度下降算法根据标注模型和实际标签信息间的差值进行自动更新;

步骤6:使用测试集测试标注模型的标注精度,判断标注精度是否达到预设阈值,若是,则执行步骤6,否则,返回步骤5继续对标注模型进行训练;

步骤7:使用训练好的电力设备标注模型进行电力设备的自动标注。

2.根据权利要求1所述的一种电力设备自动标注方法,其特征在于,所述的步骤1具体为:

采用labelimg工具对拍摄的电力设备图像进行标注,获得图像中电力设备的标签,该标签包括电力设备标注框左上角位置坐标、右下角位置坐标和电力设备类型。

3.根据权利要求1所述的一种电力设备自动标注方法,其特征在于,所述的步骤2具体为:

首先将图像的尺寸进行归一化处理,将图像归一化为300×300像素,然后保存原始图像旋转90°、180°和270°之后的图像。

4.根据权利要求1所述的一种电力设备自动标注方法,其特征在于,所述的步骤3具体为:

通过bootstrap将电力设备图像集分为训练集和测试集,训练集用于训练电力设备标注模型,测试集用于测试电力设备标注模型的性能。

5.根据权利要求1所述的一种电力设备自动标注方法,其特征在于,所述的电力设备包括杆塔、导线、防震锤、绝缘子和间隔棒。

6.根据权利要求1所述的一种电力设备自动标注方法,其特征在于,所述的电力设备标注模型为基于r-cnn的标注模型、基于fastrcnn的标注模型、基于fasterrcnn的标注模型、基于yolo的标注模型、基于sdd的标注模型或基于maskrcnn标注模型。

7.根据权利要求1所述的一种电力设备自动标注方法,其特征在于,所述的步骤5中的梯度下降算法为基于随机梯度下降的误差反向传播算法。

8.一种使用如权利要求1所述自动标注方法的电力设备自动标注装置,其特征在于,包括:

图像传感器(1),用于采集电力设备的图像;

存储器(2),用于存储电力设备原始图像、电力设备训练集图像及标签信息、电力设备测试集图像及标签信息和电力设备标注图像集标签信息;

cpu(3),内嵌电力设备自动标注方法,用于对电力设备图形进行处理,获得电力设备标注图像;

电源(4),用于为图像传感器(1)、存储器(2)和cpu(3)供电;

所述的图像传感器(1)、存储器(2)和电源(4)分别与cpu(3)相连。

9.根据权利要求8所述的一种电力设备自动标注装置,其特征在于,所述的图像传感器(1)为可见光摄像头、红外摄像头、紫外摄像头或遥感卫星。

10.根据权利要求8所述的一种电力设备自动标注装置,其特征在于,所述的电力设备自动标注装置设有gpu(5);所述的gpu(5)与cpu(3)相连。


技术总结
本发明涉及一种电力设备自动标注方法,包括获得经过设备级标注的电力设备图像集;对图像进行预处理;将电力设备图像集分为训练集和测试集;对电力设备标注模型进行训练,获得图像中电力设备的预测框和预测设备类型;采用梯度下降算法根据标注模型和实际标签信息间的差值进行自动更新;使用测试集测试标注模型的标注精度,判断标注精度是否达到预设阈值;使用训练好的电力设备标注模型进行电力设备的自动标注。自动标注装置包括图像传感器、存储器、CPU和电源,图像传感器、存储器和电源分别与CPU相连与现有技术相比,本发明具有消除拍摄角度影响,准确率高等优点。

技术研发人员:倪祺;江安烽;颜楠楠;陈宁;顾春杰;朱皇儒;黄浩;胡斌;季珊珊
受保护的技术使用者:国网上海市电力公司;国网上海能源互联网研究院有限公司
技术研发日:2020.06.15
技术公布日:2020.10.23
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1