本申请实施例涉及绝缘子检测领域,具体涉及一种绝缘子状态检测方法及绝缘子状态检测模型训练方法。
背景技术:
1、绝缘子是架空输电线路中的重要部件,同时也是电网设备的重要组成部分。它们在输电线路中可以实现电气绝缘和机械固定的作用。因此,在各种内部或外部因素造成绝缘子损坏的情况下可能导致绝缘子缺陷,尤其是电气损伤,都会导致绝缘子的工作性能下降,甚至影响整个输电线路的工作。因此,如何对每一个绝缘子进行实时缺陷检测,保证其功能的准确实现,对于电力公司来说是非常重要的。
2、目前使用单一的神经网络模型对待测图像同时识别绝缘子的位置以及对识别到的绝缘子进行分类,显然,要同时完成以上两种操作,该神经网络模型会十分复杂,而又由于缺陷绝缘子的图像样本数量较少,在这多个因素的影响下,该神经网络模型的训练效果十分一般,进而导致对图像中绝缘子缺陷的检测速率和检测准确率都很低,无法满足检测需求。
技术实现思路
1、本申请实施例提供了一种绝缘子状态检测方法及绝缘子状态检测模型训练方法,用于提升绝缘子状态检测的速率和准确率。
2、本申请实施例第一方面提供了一种绝缘子状态检测模型训练方法,所述方法包括:
3、获取第一数据集,所述第一数据集中的每组训练样本包括含有缺陷绝缘子和/或无缺陷绝缘子的绝缘子图像以及所述绝缘子图像中的绝缘子的位置信息;
4、获取初始绝缘子定位模型,将所述第一数据集输入至所述初始绝缘子定位模型,得到所述初始绝缘子定位模型输出的位置信息,当所述输出的位置信息与所述绝缘子图像中的绝缘子的位置信息之间的关系满足收敛条件时停止训练,得到目标绝缘子定位模型;
5、获取第二数据集,所述第二数据集中的每组训练样本包括第一框形图像以及所述第一框形图像的标签信息,每个所述第一框形图像显示单个绝缘子,所述第一框形图像的标签信息用于表示所述第一框形图像显示的绝缘子为无缺陷绝缘子或者缺陷绝缘子;
6、获取初始绝缘子分类模型,将所述第二数据集输入至所述初始绝缘子分类模型,得到所述初始绝缘子分类模型输出的标签信息,当所述输出的标签信息与所述第一框形图像的标签信息之间的关系满足收敛条件时停止训练,得到目标绝缘子分类模型;
7、其中,所述目标绝缘子定位模型用于识别目标图像中绝缘子的位置信息,所述目标绝缘子分类模型用于对基于所述目标图像中绝缘子的位置信息截取得到的第二框形图像进行分类,分类结果用于表示所述第二框形图像中的绝缘子为无缺陷绝缘子或者缺陷绝缘子。
8、本申请实施例第二方面提供了一种绝缘子状态检测方法,所述方法包括:
9、获取待检测的显示有绝缘子的目标图像;
10、获取预先训练完成的目标绝缘子定位模型,所述目标绝缘子定位模型由机器学习算法对第一数据集训练得到,所述第一数据集中的每组训练样本包括含有缺陷绝缘子和/或无缺陷绝缘子的绝缘子图像以及所述绝缘子图像中的绝缘子的位置信息;
11、获取预先训练完成的目标绝缘子分类模型,所述目标绝缘子分类模型由机器学习算法对第二数据集训练得到,所述第二数据集中的每组训练样本包括第一框形图像以及所述第一框形图像的标签信息,每个所述第一框形图像显示单个绝缘子,所述第一框形图像的标签信息用于表示所述第一框形图像显示的绝缘子为无缺陷绝缘子或者缺陷绝缘子;
12、将所述目标图像输入至所述目标绝缘子定位模型,以得到所述目标绝缘子定位模型输出的位置信息,基于所述目标绝缘子定位模型输出的位置信息确定包围单个绝缘子的框,截取所述包围单个绝缘子的框以得到框形图像;
13、将所述框形图像输入至所述目标绝缘子分类模型,以得到所述目标绝缘子分类模型输出的标签信息,所述目标绝缘子分类模型输出的标签信息用于表示所述框形图像显示的绝缘子为无缺陷绝缘子或者缺陷绝缘子。
14、本申请实施例第三方面提供了一种计算机设备,包括存储器和处理器,所述存储器存储有计算机程序,所述处理器执行所述计算机程序时实现前述第一方面和/或第二方面的方法。
15、本申请实施例第四方面提供了一种计算机存储介质,计算机存储介质中存储有指令,该指令在计算机上执行时,使得计算机执行前述第一方面和/或第二方面的方法。
16、从以上技术方案可以看出,本申请实施例具有以下优点:
17、目标绝缘子定位模型由机器学习算法对第一数据集训练得到,第一数据集中的每组训练样本包括含有缺陷绝缘子和/或无缺陷绝缘子的绝缘子图像以及绝缘子图像中的绝缘子的位置信息,目标绝缘子分类模型由机器学习算法对第二数据集训练得到,第二数据集中的每组训练样本包括第一框形图像以及第一框形图像的标签信息,第一框形图像的标签信息表示第一框形图像显示的绝缘子为无缺陷绝缘子或者缺陷绝缘子。由于无缺陷绝缘子的样本数量较多,因此目标绝缘子定位模型基于大量的无缺陷绝缘子样本图像和较少量的缺陷绝缘子样本图像进行训练便可获得很好的训练效果。而目标绝缘子分类模型由于仅执行分类任务,因而其网络结构相对简单,基于大量的无缺陷绝缘子样本图像和较少量的缺陷绝缘子样本图像进行训练,同样可获得很好的训练效果。因而两个模型用于绝缘子状态检测,可以保证检测的准确率和速率,满足检测需求。
1.一种绝缘子状态检测模型训练方法,其特征在于,所述方法包括:
2.根据权利要求1所述的方法,其特征在于,所述第一数据集包括第一训练集和第一验证集,所述第一训练集和所述第一验证集中的每组训练样本包括所述绝缘子图像以及所述绝缘子图像中的绝缘子的位置信息;
3.根据权利要求2所述的方法,其特征在于,所述方法还包括:
4.根据权利要求1所述的方法,其特征在于,所述第一数据集中每组训练样本还包括所述绝缘子图像中每个绝缘子的标签信息,所述绝缘子图像中绝缘子的标签信息用于表示所述绝缘子图像中的绝缘子为无缺陷绝缘子或者缺陷绝缘子;
5.根据权利要求1所述的方法,其特征在于,所述第二数据集包括第二训练集和第二验证集,所述第二训练集和所述第二验证集中的每组训练样本包括所述第一框形图像以及所述第一框形图像的标签信息;
6.根据权利要求5所述的方法,其特征在于,所述方法还包括:
7.根据权利要求1至6任一项所述的方法,其特征在于,所述方法还包括:
8.根据权利要求1至6任一项所述的方法,其特征在于,所述方法还包括:
9.一种绝缘子状态检测方法,其特征在于,所述方法包括:
10.一种计算机设备,包括存储器和处理器,所述存储器存储有计算机程序,其特征在于,所述处理器执行所述计算机程序时实现如权利要求1至9中任一项所述的方法。
11.一种计算机存储介质,其特征在于,所述计算机存储介质中存储有指令,所述指令在计算机上执行时,使得所述计算机执行如权利要求1至9中任一项所述的方法。