[发明专利]一种航拍绝缘子目标检测方法及装置

权利要求书

1.一种航拍绝缘子目标检测方法，其特征在于，包括：

获取待测航拍图像；

将所述取待测航拍图像输入进训练好的基于FasterR-CNN的航拍静态目标识别注意力模型中，识别航拍图像中的绝缘子目标；所述基于FasterR-CNN的航拍静态目标识别注意力模型中包括ResNet101网络以及FPN网络，在所述ResNet101网络中引入SGE注意力模块；

具体为：

将ResNet作为基本骨干网络；

ResNet包括依次连接的卷积层C1-C5；卷积层C2和C3之间，C3和C4之间，C4和C5之间嵌入有所述SGE注意力模块。

2.根据权利要求1所述的航拍绝缘子目标检测方法，其特征在于，在所述获取待测航拍图像，之后还包括：

将所述待测航拍图像转化成预置尺寸的图像。

3.根据权利要求1所述的航拍绝缘子目标检测方法，其特征在于，在所述将所述取待测航拍图像输入进训练好的基于FasterR-CNN的航拍静态目标识别注意力模型中，识别航拍图像中的绝缘子目标，之前还包括：

获取绝缘子航拍数据集；

将所述绝缘子航拍数据集中的航拍图像标准化成统一尺寸；

标注所述航拍图像中的绝缘子目标；

将标注好的所述航拍图像分成训练集和测试集，用于训练所述基于Faster R-CNN的航拍静态目标识别注意力模型。

4.根据权利要求1所述的航拍绝缘子目标检测方法，其特征在于，在所述将所述取待测航拍图像输入进训练好的基于FasterR-CNN的航拍静态目标识别注意力模型中，识别航拍图像中的绝缘子目标，包括：

采用引入所述SGE注意力模块的所述ResNet101网络对所述待测航拍图像进行特征提取，获得不同尺寸的特征图；

将不同尺寸的特征图输入进区域生成网络FPN中，得到多个包含绝缘子目标的候选框；

通过RoIAlign将多个包含绝缘子目标的候选框进行尺度统一；

在候选框区域上进行类别判断以及矩形框修正，得到绝缘子目标的识别结果。

5.一种航拍绝缘子目标检测装置，其特征在于，包括：

第一获取单元，用于获取待测航拍图像；

第一识别单元，用于将所述取待测航拍图像输入进训练好的基于Faster R-CNN的航拍静态目标识别注意力模型中，识别航拍图像中的绝缘子目标；所述基于FasterR-CNN的航拍静态目标识别注意力模型中包括ResNet101网络以及FPN网络，在所述ResNet101网络中引入SGE注意力模块；

具体为：

将ResNet作为基本骨干网络；

ResNet包括依次连接的卷积层C1-C5；卷积层C2和C3之间，C3和C4之间，C4和C5之间嵌入有所述SGE注意力模块。

6.根据权利要求5所述的航拍绝缘子目标检测装置，其特征在于，还包括：

第一尺寸转化单元，用于将所述待测航拍图像转化成预置尺寸的图像。

7.根据权利要求5所述的航拍绝缘子目标检测装置，其特征在于，还包括：

第二获取单元，用于获取绝缘子航拍数据集；

第二尺寸转化单元，用于将所述绝缘子航拍数据集中的航拍图像标准化成统一尺寸；

标注单元，用于标注所述航拍图像中的绝缘子目标；

训练单元，用于将标注好的所述航拍图像分成训练集和测试集，用于训练所述基于FasterR-CNN的航拍静态目标识别注意力模型。

8.根据权利要求5所述的航拍绝缘子目标检测装置，其特征在于，所述第一识别单元还包括：

特征提取单元，用于将不同尺寸的特征图输入进区域生成网络FPN中，得到多个包含绝缘子目标的候选框；

尺度统一单元，用于通过RoIAlign将多个包含绝缘子目标的候选框进行尺度统一；

第二识别单元，用于在候选框区域上进行类别判断以及矩形框修正，得到绝缘子目标的识别结果。