[发明专利]一种基于关联规则学习的图像检索方法、装置及存储介质

说明书

本发明公开了一种基于关联规则学习的图像检索方法、装置及存储介质，通过获取待检索图像和检索数据库中图像；基于关联规则学习算法提取待检索图像和检索数据库中图像的局部相关性特征；对待检索图像和检索数据库中图像的局部相关性特征进行特征匹配，确定检索数据库中和待检索图像属于同一子类别的检索结果。由此，该图像检索方法通过关联规则学习算法，对图像内部的关联信息进行挖掘，加强了细粒度特征表示，从而能够增强不同子类间判别性和区分度。

技术领域

本发明涉及图像检索技术领域，具体涉及一种基于关联规则学习的图像检索方法、装置及存储介质。

背景技术

电网图像检索旨在用户向计算机输入一个查询对象，计算机返回与该查询对象同属于同一子类别的检索结果。随着深度学习和人工智能的日益崛起，电网图像检索有着广阔的发展和应用前景。例如，可以精准地检索出不同子类的违章行为，从而帮助业务员提升电力作业的自动化和精准化。

目前，电网图像检索所面临的挑战：不同子类间差异较小，同一子类内方差较大。这与我们所希望的“子类间差异大，子类内方差小”恰恰背道而驰。由此，如何增强不同子类别间区分度和判别性是目前亟需解决的问题。

发明内容

有鉴于此，本发明实施例提供了涉及一种基于关联规则学习的图像检索方法、装置及存储介质，以解决现有技术中在进行电网图像检索时不同子类间差异较小的技术问题。

本发明提出的技术方案如下：

本发明实施例第一方面提供一种基于关联规则学习的图像检索方法，包括：获取待检索图像和检索数据库中图像；基于关联规则学习算法提取待检索图像和检索数据库中图像的局部相关性特征；对待检索图像和检索数据库中图像的局部相关性特征进行特征匹配，确定检索数据库中和待检索图像属于同一子类别的检索结果。

可选地，基于关联规则学习算法提取待检索图像和检索数据库中图像的局部相关性特征，包括：基于关联规则学习算法提取待检索图像的局部相关性特征；基于关联规则学习算法提取检索数据库中每一张图像的局部相关性特征。

可选地，基于关联规则学习算法提取待检索图像的局部相关性特征，包括：提取待检索图像的特征图和每张特征图中的激活像素；基于所述特征图和每张特征图中的激活像素，采用关联规则学习算法计算待检索图像的局部相关性特征。

可选地，提取待检索图像的特征图和每张特征图中的激活像素，包括：采用预训练模型将待检索图像映射为一组特征图；计算每张特征图中所有像素点响应值的均值；比较每张特征图中所有像素点的响应值和相应均值的大小，将响应值大于相应均值的像素点作为特征图中的激活像素。

可选地，基于所述特征图和每张特征图中的激活像素，采用关联规则学习算法计算待检索图像的局部相关性特征，包括：基于所述特征图和每张特征图中的激活像素生成特征图集合和激活像素集；根据所述特征图集合和激活像素集计算待检索图像的频繁模式；采用池化操作提取所述频繁模式中的特征作为待检索图像的局部相关性特征。

可选地，对待检索图像和检索数据库中图像的局部相关性特征进行特征匹配，确定检索数据库中和待检索图像属于同一子类别的检索结果，包括：计算待检索图像的局部相关性特征和检索数据库中每一张图像的局部相关性特征的相似度，得到多个相似度；对多个相似度进行排序，提取前预设个数的相似度对应的图像作为与待检索图像属于同一子类别的检索结果。

可选地，所述相似度采用余弦相似度函数计算。

本发明实施例第二方面提供一种基于关联规则学习的图像检索装置，包括：图像获取模块，用于获取待检索图像和检索数据库中图像；特征提取模块，用于基于关联规则学习算法提取待检索图像和检索数据库中图像的局部相关性特征；特征匹配模块，用于对待检索图像和检索数据库中图像的局部相关性特征进行特征匹配，确定检索数据库中和待检索图像属于同一子类别的检索结果。