DL图像检索

本节内容包含：

模块原理

DL图像检索是一种利用图像语义信息（如颜色、纹理、布局等）进行分析和检索的技术。从功能上来说，它与DL分类类似，把整张图像归为若干类别中的某一类，在物体识别与分拣上有广泛的应用。从算法本质上来说存在较大的区别：

  • DL图像分类：分类模型从图像中提取到图像特征，直接通过内嵌分类器输出图像类别。该算法适用于类间差异较大、类别固定的图像场景。

  • DL图像检索：检索模型从图像中提取到图像特征，并无内置分类器，不进行直接分类，而是通过与数据库(Gallery）中所有图像特征进行相似性搜索，将库中的图像按照相似度大小进行排列，输出指定的TopN检索结果。该算法适用于类内差异较小、类别需新增的图像场景。

图 1 分类和检索应用场景区别

使用方法

  1. 拖拽图像源模块至流程编辑区，并输入图像；

  2. 拖拽DL图像检索模块至流程编辑区，双击该模块进行参数配置，加载训练好的图像检索模型文件以及训练得到的Gallery文件；

  3. 点击执行即可输出检索结果，如下图所示。

图 2 DL图像检索示例效果图

参数配置

以下仅介绍该模块的运行参数详情。通过配置运行参数，可定义当前模块如何处理输入的数据。

说明：
模型文件路径

自带默认模型文件，也可根据实际需求自行添加训练好的模型文件。

方案存模型

使能后，将模型数据保存到方案文件或流程文件中，跨机加载方案时无需再次输入模型文件路径。

Gallery路径

Gallery文件在模型训练时与模型文件一同生成，路径为深度学习训练工具软件的安装路径下的DeepLearningModel文件夹下。

最小相似度

当待输出的类别与输入图像的相似度小于该参数值时，该类别将不再被输出。

查询类别

点击查询类别后弹框界面如下图所示。文本框中会显示当前Gallery中总类别数，点击右侧可以直接导出所有类别信息。在文本框中输入待查询的类别名称后，下方会自动匹配相关类别的缩略图信息、类别名称以及Gallery中已注册的图像个数。

图 3 查询类别
注册图像

点击注册图像后弹框界面如下图所示。

图 4 注册图像

注册方式分为按图像注册和按文件夹注册两种。

  • 按图像注册：首先输入要注册的类别名称，点击可以添加待注册的图像，可以一次性选入多个，也可分批次选择图像，再点击注册即可注册所有添加的图像。

  • 按文件夹注册：点击选择文件夹后选择文件夹，此时默认将文件夹名称作为类别名称进行注册，也支持对类别名称进行修改。同时也应注意，一个类别最多可注册100张图片。此外，按文件夹注册支持一次注册多个文件夹，即同时注册多个类别。

模块结果

该模块的模块结果介绍请见DL图像检索。