[发明专利]图像信息提取方法、训练方法及装置、介质和电子设备

说明书

本公开提供了一种图像信息提取方法、图像信息提取网络的训练方法、图像信息提取装置、图像信息提取网络的训练装置、计算机可读存储介质和电子设备，涉及图像处理技术领域。该图像信息提取方法包括：获取图像，提取图像的局部描述子；计算局部描述子的置信度；基于局部描述子以及局部描述子的置信度，提取图像的全局描述子。本公开可以提高提取图像描述子的精确度。

技术领域

本公开涉及图像处理技术领域，具体而言，涉及一种图像信息提取方法、图像信息提取网络的训练方法、图像信息提取装置、图像信息提取网络的训练装置、计算机可读存储介质和电子设备。

背景技术

随着终端技术的发展，视觉导航、AR(Augmented Reality，增强现实)、场景识别、无人驾驶等功能均可以基于终端来实现。为了实现这些技术，终端通常需要对图像进行一系列处理。而在对图像进行处理的过程中，特征描述子的提取尤为重要，直接影响到上层应用是否能够顺利实现。

对图像的描述可以基于全局描述子和局部描述子来实现，在例如视觉导航、图像检索等技术中，为了更准确的表达出图像，需要同时利用到全局描述子和局部描述子。

目前，在分别提取全局描述子和局部描述子的方案中，未考虑到全局描述子与局部描述子的相关性，导致全局描述子和局部描述子对图像的表征能力较差，提取到的描述子不够精确，造成图像处理的效果不理想。

发明内容

本公开提供一种图像信息提取方法、图像信息提取网络的训练方法、图像信息提取装置、图像信息提取网络的训练装置、计算机可读存储介质和电子设备，进而至少在一定程度上克服提取的描述子不精确而导致图像处理效果不佳的问题。

根据本公开的第一方面，提供了一种图像信息提取方法，包括：获取图像，提取图像的局部描述子；计算局部描述子的置信度；基于局部描述子以及局部描述子的置信度，提取图像的全局描述子。

根据本公开的第二方面，提供了一种图像信息提取网络的训练方法，包括：获取图像信息提取网络和训练集，其中，训练集包括多个图像对，每个图像对包括属于同一场景的第一训练图像和第二训练图像；将第一训练图像输入图像信息提取网络，得到第一训练图像的局部描述子和全局描述子；将第二训练图像输入图像信息提取网络，得到第二训练图像的局部描述子和全局描述子；根据第一训练图像的局部描述子和全局描述子以及第二训练图像的局部描述子和全局描述子，利用损失函数训练图像信息提取网络。

根据本公开的第三方面，提供了一种图像信息提取装置，包括：第一提取模块，用于获取图像，提取图像的局部描述子；置信度计算模块，用于计算局部描述子的置信度；第二提取模块，用于基于局部描述子以及局部描述子的置信度，提取图像的全局描述子。

根据本公开的第四方面，提供了一种图像信息提取网络的训练装置，包括：网络获取模块，用于获取图像信息提取网络和训练集，其中，训练集包括多个图像对，每个图像对包括属于同一场景的第一训练图像和第二训练图像；第一训练数据确定模块，用于将第一训练图像输入图像信息提取网络，得到第一训练图像的局部描述子和全局描述子；第二训练数据确定模块，用于将第二训练图像输入图像信息提取网络，得到第二训练图像的局部描述子和全局描述子；训练模块，用于根据第一训练图像的局部描述子和全局描述子以及第二训练图像的局部描述子和全局描述子，利用损失函数训练图像信息提取网络。

根据本公开的第五方面，提供了一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时实现上述的图像信息提取方法或图像信息提取网络的训练方法。

根据本公开的第六方面，提供了一种电子设备，包括处理器；存储器，用于存储一个或多个程序，当一个或多个程序被处理器执行时，使得所述处理器实现上述的图像信息提取方法或图像信息提取网络的训练方法。