[发明专利]一种用于覆冰图像细粒度分类的图像特征提取方法和装置

说明书

本发明公开了一种用于覆冰图像细粒度分类的图像特征提取方法和装置，所述方法包括如下步骤：步骤1，对输入图像进行预处理；步骤2，对预处理的图像进行特征提取。本发明在图像细粒度分类时考虑充分建模各层之间的信息传递，有效地提取图像特征，能对包含相近类别物体的图像进行更准确分类。

技术领域

本发明涉及深度学习技术领域，尤其涉及一种用于覆冰图像细粒度分类的图像特征提取方法的装置。

背景技术

近年来，随着全球气温逐渐变暖，极端天气的不断增多，电网系统面临巨大的挑战。当前，我国电力系统在应对极端天气方面还存在很多技术欠缺，如果不能及时对可能出现的极端天气进行有效的预测和干预，可能造成灾难性的后果。冰风灾害是供电系统面临的一大难题。世界各地都曾出现过不同程度的冰风灾害，我国电网系统也发生过多次冰风灾害，而且这种灾害的发生频率呈现上升趋势。2008年年初，我国南方部分省市遭受了前所未有的低温雨雪冰冻灾害，冰冻灾害造成国家电网公司的直接财产损失达到104.5亿元，而灾后的电网恢复重建和改造投入资金达390亿元，经济损失巨大。巨大的经济损失和社会影响警示我们，开展电力系统防灾研究、保障电力系统在冰风灾害时的安全稳定运行是当前刻不容缓的挑战和艰巨的任务。在应对冰风灾害时除了采取防覆冰技术、融冰技术和除冰技术之外，对冰风灾害进行有效的识别和预测也具有十分重要的意义，而利用视觉图片判断电力设施是否覆冰是一个非常有效的方法。

图像细粒度分类是在相同大类中更细致的区分不同小类，如区分覆冰与非覆冰图像。因为大类物体下不同小类的物体具有相近的结构，这导致了区分不同小类物体图像的难度大幅度上升，使得准确区分相同大类下的不同小类物体具有挑战性。此外由于环境光照、拍摄设备、物体角度、运动状态、拍摄距离等条件的不同导致图像质量不同，物体在图像中展现的姿态也不同，使得图像细粒度分类的问题更加复杂。在覆冰与非覆冰两种图像中物体的外观具有很高的相似性，在雨雪天气中拍摄的图像则更加容易与覆冰图像混淆，这使得区分覆冰与非覆冰图像的难度大幅提升。

目前存在的图像细粒度检测识别方法大多基于图像的纹理、空间信息。有人提出了提前标注训练图像中待识别物体的位置来过滤背景因素的影响以达到更好的分类结果，但是这需要大量的人工标注，对训练数据集需要更高的要求。还有人通过将训练图像缩放，在多个不同尺度图像上分类训练模型，最后通过投票的方式决定物体类别，但是这种方法受需要额外的时间和空间来训练模型。有人提出利用自主学习的方式让模型选择放大图像中的哪些部分以达到更好的结果，但是如果训练数据中物体的位置多种多样将导致结果变差。

上述方法在某些前提假设下有较好的分类结果，但对于较为一般的图像分类结果有所下降，因此需要一种对数据集要求更少的方法提取图像的特征并与其他类别区分开。

发明内容

针对现有技术的不足，本发明提供了一种用于覆冰图像细粒度分类的图像特征提取方法和装置，其中所述方法包括如下步骤：

步骤1，对输入的覆冰图像进行预处理；

步骤2，对预处理的图像进行特征提取。

步骤1包括：对输入的覆冰图像进行缩放(缩放到256*256像素大小的图像)，裁剪(保留中心224*224像素大小的图像区域)、归一化(减去均值后除以标准差，此处均值及标准差均为ImageNet数据集的均值和标准差，可以参考：http://image-net.org/index)的预处理，得到预设分辨率的图像，并将其按照三原色RGB三通道的顺序组成输入矩阵。

步骤2包括：对预处理后的图像即步骤1得到的输入矩阵进行卷积(可以参考：https://pytorch.org/docs/stable/nn.html#conv2d)、注意力机制模块Attention和池化Pooling操作得到输出特征图，具体包括如下步骤：