[发明专利]图像特征提取模型的训练方法和图像检索方法

权利要求书

1.一种图像特征提取模型的训练方法，包括：

获取样本图像，以及所述样本图像对应的类型；

将所述样本图像输入图像特征提取模型，以生成所述样本图像对应的图像特征；其中，所述图像特征提取模型包括卷积神经网络；

根据所述样本图像对应的图像特征在特征空间中的距离，对所述卷积神经网络进行训练；以及

当相同类型的所述样本图像对应的图像特征，在所述特征空间中的位置重合，并且不同类型的所述样本图像对应的图像特征，在所述特征空间中均匀分布时，完成对所述图像特征提取模型的训练。

2.如权利要求1所述的训练方法，其中，所述根据所述样本图像对应的图像特征在特征空间中的距离，对所述卷积神经网络进行训练，包括：

调整所述卷积神经网络的参数，以减小相同类型的所述样本图像对应的图像特征在所述特征空间中的距离。

3.如权利要求2所述的训练方法，其中，所述根据所述样本图像对应的图像特征在特征空间中的距离，对所述卷积神经网络进行训练，还包括：

调整所述卷积神经网络的参数，以增大不同类型的所述样本图像对应的图像特征在所述特征空间中的距离。

4.如权利要求3所述的训练方法，其中，所述根据所述样本图像对应的图像特征在特征空间中的距离，对所述卷积神经网络进行训练，还包括：

调整所述卷积神经网络的参数，以使第一类型的所述样本图像对应的图像特征和第二类型的所述样本图像对应的图像特征在所述特征空间中的距离，与第一类型的所述样本图像对应的图像特征和第三类型的所述样本图像对应的图像特征在所述特征空间中的距离的差值小于或者等于预设阈值。

5.一种图像检索方法，包括：

获取待检索图像；

将所述待检索图像输入如权利要求1-4所述的训练方法训练后的图像特征提取模型，以生成所述待检索图像对应的图像特征；

确定所述待检索图像对应的图像特征和图像库中图像对应的图像特征在特征空间中的距离；以及

根据所述距离，确定所述图像库中与所述待检索图像类型相同的图像。

6.如权利要求5所述的方法，其中，在所述根据所述距离，确定所述图像库中与所述待检索图像类型相同的图像之后，还包括：将所述图像库中与所述待检索图像类型相同的图像，设置在图像检索结果列表的顶端。

7.如权利要求6所述的方法，其中，所述将所述图像库中与所述待检索图像类型相同的图像，设置在图像检索结果列表的顶端，包括：

确定所述待检索图像对应的图像特征和所述图像库中每种类型的图像对应的图像特征在所述特征空间中的距离；

按照所述距离从小到大的顺序，将所述图像库中每种类型的图像从上到下设置在所述图像检索结果列表中。

8.一种图像特征提取模型的训练装置，包括：

第一获取模块，用于获取样本图像，以及所述样本图像对应的类型；

第一输入模块，用于将所述样本图像输入图像特征提取模型，以生成所述样本图像对应的图像特征；其中，所述图像特征提取模型包括卷积神经网络；

训练模块，用于根据所述样本图像对应的图像特征在特征空间中的距离，对所述卷积神经网络进行训练；以及

完成模块，用于当相同类型的所述样本图像对应的图像特征，在所述特征空间中的位置重合，并且不同类型的所述样本图像对应的图像特征，在所述特征空间中均匀分布时，完成对所述图像特征提取模型的训练。

9.如权利要求8所述的训练装置，其中，所述训练模块，包括：

第一调整子模块，用于调整所述卷积神经网络的参数，以减小相同类型的所述样本图像对应的图像特征在所述特征空间中的距离。

10.如权利要求9所述的训练装置，其中，所述训练模块，还包括：

第二调整子模块，用于调整所述卷积神经网络的参数，以增大不同类型的所述样本图像对应的图像特征在所述特征空间中的距离。