输电线路下垫面环境分类及模型训练方法、装置及设备与流程

本发明涉及计算机视觉，尤其涉及一种输电线路下垫面环境分类及模型训练方法、装置及设备。

背景技术：

1、在电网应用中，对输电线路下垫面环境进行分类识别，对于输电线路布局规划是至关重要的。

2、传统的环境分类方法一般分为光谱指数法和浅层机器学习法。光谱指数法，根据不同地物反射光谱差异，通过波段运算构建反应特定地物属性的光谱指数，进行下垫面分类。光谱指数方法原理简单，计算高效，但是没有考虑复杂环境下地物光谱多变性，导致地物识别精度通常较低，且分割阈值通常难以确定，无法满足对输电线路环境检测的需求。浅层机器学习法，通过设计手工特征，如光谱特征、纹理特征等，根据各类别训练样本特性构建判别函数，优化判别边界，利用训练好的分类器进行分类。该类方法更够有效识别各类地物，是目前应用最为广泛的方法。然而，上述方法过度依赖手工设计特征和先验知识，难以挖掘表征地物本质差异的高层语义特征，泛化能力有限，在复杂环境下的鲁棒性不足，难以实现输电线路下垫面精细化分类。

技术实现思路

1、本发明提供了一种输电线路下垫面环境分类及模型训练方法、装置及设备，以实现高识别准确率及效率的输电线路下垫面环境分类模型生成。

2、根据本发明的一方面，提供了一种输电线路下垫面环境分类模型训练方法，该方法包括：

3、采集输电线路环境影像，并对所述输电线路环境影像进行标注，得到输电线路环境分类数据集；

4、将所述输电线路环境分类数据集划分为输电线路下垫面环境分类模型训练集和测试集；

5、构建耦合注意力机制的二维卷积神经网络模型2dcnn-cbam作为初始输电线路下垫面环境分类模型，并采用所述训练集对所述初始输电线路下垫面环境分类模型进行模型训练；

6、其中，所述初始输电线路下垫面环境分类模型，包括：二维卷积层、激活层、注意力机制层、最大池化层以及全连接层；

7、根据所述测试集对训练结果进行评价，当评价通过时得到输电线路下垫面环境分类模型。

8、根据本发明的另一方面，提供了一种输电线路下垫面环境分类方法，包括：

9、获取待识别输电线路环境影像；

10、将所述待识别输电线路环境影像输入至如本发明任一实施例所提供的输电线路下垫面环境分类模型训练方法，训练得到的输电线路下垫面环境分类模型中，得到输电线路下垫面环境分类结果。

11、根据本发明的另一方面，提供了一种输电线路下垫面环境分类模型训练装置，该装置包括：

12、数据集生成模块，用于采集输电线路环境影像，并对所述输电线路环境影像进行标注，得到输电线路环境分类数据集；

13、数据集划分模块，用于将所述输电线路环境分类数据集划分为输电线路下垫面环境分类模型训练集和测试集；

14、模型训练模块，用于构建耦合注意力机制的二维卷积神经网络模型2dcnn-cbam作为初始输电线路下垫面环境分类模型，并采用所述训练集对所述初始输电线路下垫面环境分类模型进行模型训练；

15、其中，所述初始输电线路下垫面环境分类模型，包括：二维卷积层、激活层、注意力机制层、最大池化层以及全连接层；

16、模型评价模块，用于根据所述测试集对训练结果进行评价，当评价通过时得到输电线路下垫面环境分类模型。

17、根据本发明的另一方面，提供了一种输电线路下垫面环境分类装置，该装置包括：

18、环境影像获取模块，用于获取待识别输电线路环境影像；

19、环境分类结果确定模块，用于将所述待识别输电线路环境影像输入至如本发明任一实施例所提供的输电线路下垫面环境分类模型训练方法，训练得到的输电线路下垫面环境分类模型中，得到输电线路下垫面环境分类结果。

20、根据本发明的另一方面，提供了一种电子设备，所述电子设备包括：

21、至少一个处理器；以及

22、与所述至少一个处理器通信连接的存储器；其中，

23、所述存储器存储有可被所述至少一个处理器执行的计算机程序，所述计算机程序被所述至少一个处理器执行，以使所述至少一个处理器能够执行本发明任一实施例所述的输电线路下垫面环境分类模型训练方法，或者输电线路下垫面环境分类方法。

24、根据本发明的另一方面，提供了一种计算机可读存储介质，所述计算机可读存储介质存储有计算机指令，所述计算机指令用于使处理器执行时实现本发明任一实施例所述的输电线路下垫面环境分类模型训练方法，或者输电线路下垫面环境分类方法。

25、本发明实施例的技术方案，通过采集输电线路环境影像，并对所述输电线路环境影像进行标注，得到输电线路环境分类数据集；将所述输电线路环境分类数据集划分为输电线路下垫面环境分类模型训练集和测试集；构建耦合注意力机制的二维卷积神经网络模型作为初始输电线路下垫面环境分类模型，并采用所述训练集对所述初始输电线路下垫面环境分类模型进行模型训练；其中，所述初始输电线路下垫面环境分类模型，包括：二维卷积层、激活层、注意力机制层、最大池化层以及全连接层；根据所述测试集对训练结果进行评价，当评价通过时得到输电线路下垫面环境分类模型，解决了复杂环境下输电线路下垫面地物的精细化分类问题，通过构建耦合注意力机制的网络模型，实现高识别准确率及效率的输电线路下垫面环境分类模型生成。

26、应当理解，本部分所描述的内容并非旨在标识本发明的实施例的关键或重要特征，也不用于限制本发明的范围。本发明的其它特征将通过以下的说明书而变得容易理解。

技术特征：

1.一种输电线路下垫面环境分类模型训练方法，其特征在于，包括：

2.根据权利要求1所述的方法，其特征在于，构建耦合注意力机制的二维卷积神经网络模型2dcnn-cbam作为初始输电线路下垫面环境分类模型，包括：

3.根据权利要求2所述的方法，其特征在于，将二维卷积层、激活层、以及注意力机制层连接组成输电线路下垫面环境分类卷积网络层，包括：

4.根据权利要求3所述的方法，其特征在于，将二维卷积层、激活层、以及由通道注意力层cam和空间注意力层sam构成的注意力机制层，连接组成输电线路下垫面环境分类卷积网络层，包括：

5.根据权利要求1所述的方法，其特征在于，根据所述测试集对训练结果进行评价，包括：

6.一种输电线路下垫面环境分类方法，其特征在于，包括：

7.一种输电线路下垫面环境分类模型训练装置，其特征在于，包括：

8.一种输电线路下垫面环境分类装置，其特征在于，包括：

9.一种电子设备，其特征在于，所述电子设备包括：

10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质存储有计算机指令，所述计算机指令用于使处理器执行时实现权利要求1-5中任一项所述的输电线路下垫面环境分类模型训练方法，或者权利要求6所述的输电线路下垫面环境分类方法。

技术总结
本发明实施例公开了一种输电线路下垫面环境分类及模型训练方法、装置及设备。模型训练方法包括：采集输电线路环境影像并进行标注，得到输电线路环境分类数据集；将输电线路环境分类数据集划分为输电线路下垫面环境分类模型训练集和测试集；构建耦合注意力机制的二维卷积神经网络模型作为初始输电线路下垫面环境分类模型，采用训练集对初始输电线路下垫面环境分类模型进行模型训练；初始输电线路下垫面环境分类模型，包括：二维卷积层、激活层、注意力机制层、最大池化层及全连接层；根据测试集对训练结果进行评价，得到输电线路下垫面环境分类模型。通过构建耦合注意力机制的网络模型，实现高识别准确率及效率的输电线路下垫面环境分类模型生成。

技术研发人员：吴亮,罗兵,肖微,许国伟,张豪峰,林昌松,李欣,陈梓荣,纪长城,翟晗,林来鑫,杜欢,陈鸿适,杜金亮,范晟,蔡哲淳
受保护的技术使用者：广东电网有限责任公司
技术研发日：
技术公布日：2024/1/15