[发明专利]一种图像检索方法、装置及存储介质

权利要求书

1.一种图像检索方法，其特征在于，包括：

根据用户的样本图像，标记图像的轮廓和类别；

基于yolact++实例分割模型和yolov5s目标检测模型，对已标记的图像进行训练，生成样本图像训练模型；

根据样本图像训练模型，建立样本图像特征索引数据库；

提取待检索图像的特征信息，并在所述样本图像特征索引数据库中进行匹配，生成检索结果。

2.根据权利要求1所述的图像检索方法，其特征在于，所述根据用户的样本图像，标记图像的轮廓和类别的步骤，包括：

获取用户的样本图像；

根据样本图像，确定样本图像中的物品类别；

根据物品类别，标记样本图像中的物品轮廓。

3.根据权利要求1所述的图像检索方法，其特征在于，所述基于yolact++实例分割模型和yolov5s目标检测模型，对已标记的图像进行训练，生成样本图像训练模型的步骤，包括：

基于yolact++实例分割模型和已标记的图像，生成实例分割模型M1；

基于yolov5s目标检测模型和已标记的图像，生成目标检测模型M2。

4.根据权利要求3所述的图像检索方法，其特征在于，所述根据样本图像训练模型，建立样本图像特征索引数据库的步骤，包括：

根据实例分割模型，分割待建的索引图像中的要素，获得包含要素的第一图像；

根据目标检测模型，确定第一图像中的要素的类别信息；

对所述第一图像进行预处理，得到第二图像，所述第二图像包含要素和边缘背景；

对所述第二图像进行二值化处理，得到第三图像，所述第三图像的要素突出于边缘背景；

根据第二图像获取待建的索引图像中的颜色特征；

根据第二图像获取待建的索引图像中的纹理特征；

根据第三图像获取待建的索引图像中的轮廓特征；

将所述颜色特征、纹理特征、轮廓特征和类别信息聚合存入所述样本图像索引数据库。

5.根据权利要求1所述的图像检索方法，其特征在于，所述提取待检索图像的特征信息，并在所述样本图像特征索引数据库中进行匹配，生成检索结果的步骤，包括：

提取待检索图像的特征信息，所述特征信息包括所述颜色特征、纹理特征、轮廓特征和类别信息；

将待检索图像的类别信息与样本图像索引数据库的对应类别信息进行匹配，确定第一索引子库；

根据待检索图像的轮廓特征、纹理特征和颜色特征对第一索引子库进行进一步筛选，确定第二索引子库。

6.一种图像检索装置，其特征在于，包括：

标记模块，用于根据用户的样本图像，标记图像的轮廓和类别；

第一生成模块，用于基于yolact++实例分割模型和yolov5s目标检测模型，对已标记的图像进行训练，生成样本图像训练模型；

建立模块，用于根据样本图像训练模型，建立样本图像特征索引数据库；

第二生成模块，用于提取待检索图像的特征信息，并在所述样本图像特征索引数据库中进行匹配，生成检索结果。