[发明专利]一种对图像中的绝缘子的定位方法

权利要求书

1.一种对图像中的绝缘子的定位方法，其特征在于包括以下步骤：

1)采集绝缘子图像；

2)采用色彩抖动法、叠加噪声法、裁切混淆法和加权混淆法对图像进行扩充；

3)利用labelImg工具对所有无人机采集图像及扩充图像中绝缘子的位置进行标注；

4)构建图像训练集和测试集；

5)利用训练集进行基于改进SSD算法的绝缘子目标模型检测训练；

6)利用测试集对完成训练的模型系统进行测试；

7)测试通过后，输入实际采集的绝缘子图像进行精确定位分辨。

2.根据权利要求1所述的一种对图像中的绝缘子的定位方法，其特征在于：步骤2)中，所述的色彩抖动法为通过调整图像的亮度、饱和度和对比度，进行图像的颜色增强。

3.根据权利要求1所述的一种对图像中的绝缘子的定位方法，其特征在于：步骤2)中，所述的叠加噪声法是在绝缘子图像上叠加高斯噪声和椒盐噪声。

4.根据权利要求1所述的一种对图像中的绝缘子的定位方法，其特征在于：步骤2)中，所述的裁切混淆法是对随机选取绝缘子训练集中的A和B两张图像，随机裁切样本A中一部分矩阵区域C，并保证该矩形区域小于B图像，再将矩形区域C随机覆盖到图像B指定区域上。

5.根据权利要求1所述的一种对图像中的绝缘子的定位方法，其特征在于：步骤2)中，所述的加权混淆法是对两张绝缘子图像进行加权求和，生成新的绝缘子样本图像，图像加权混淆公式如下：

其中(p_i,q_i)和(p_j,q_j)是从绝缘子训练数据中随机抽取的两个样本，p为绝缘子图像，q为该图像的标签，λ∈[0,1]，该参数定义了两个样本进行加权求和时各自的权重。

6.根据权利要求1所述的一种对图像中的绝缘子的定位方法，其特征在于：步骤5)中，所述的改进SSD算法包括以下步骤：

501)对SSD算法中目标检测特征图f进行两倍上采样后与目标检测特征图e进行融合，作为新的目标检测特征图e’

502)将目标检测特征图e’进行两倍上采样后与目标检测特征图d特征图融合，作为新的目标检测特征图d’

503)将目标检测特征图d’进行两倍上采样后与目标检测特征图c融合，作为新的目标检测特征图c’

504)将目标检测特征图c’进行两倍上采样后与目标检测特征图b融合，作为新的目标检测特征图b’

505)将目标检测特征图b’进行两倍上采样后与目标检测特征图a融合，作为新的目标检测特征图a’

506)在特征图a’、b’、c’、d’、e’、f’上进行目标检测。