[发明专利]图像检索模型的训练方法、图像检索方法、计算机设备

摘要

本发明公开了图像检索模型的训练方法、图像检索方法、存储介质和计算机设备。训练方法包括：获取训练样本集，训练样本集包括用户拍摄场景图片集和高清广告场景图片集；构建特征提取网络和属性分类器；将训练样本集中的图片输入特征提取网络，以对特征提取网络进行训练，并输出训练特征向量集；将特征向量集输入至属性分类器，以对属性分类器进行训练。检索方法包括：将待检索图像和图像库中的图像分别输入进图像检索模型，图像检索模型分别输出待检索特征向量以及图像库特征向量集；计算进行哈希编码后的待检索特征向量与图像库特征向量集中各个特征向量的汉明距离值；按照汉明距离值从小至大的顺序将图像库中的图像排序。

主权项

1.一种图像检索模型的训练方法，其特征在于，包括：获取训练样本集，所述训练样本集包括用户拍摄场景图片集和高清广告场景图片集；构建基于深度卷积神经网络的图像检索模型，所述图像检索模型包括特征提取网络和属性分类器；将所述训练样本集中的图片输入所述特征提取网络，以对所述特征提取网络进行训练，并通过所述特征提取网络输出训练特征向量集；将所述特征向量集输入至所述属性分类器，以对所述属性分类器进行训练。