摘要
本发明公开了一种基于人脸卷积神经网络和随机kd#树森林的图像检索方法,包括如下步骤:采集人脸图片数据集;对人脸图片数据集采用卷积神经网络模型方法进行训练;训练后得到一个卷积神经网络模型;对原始人脸图片数据集采用已训练的卷积神经网络模型提取特征向量;根据生成的特征向量及其对应的图片标号,构造随机kd#树森林索引;对于每一张需要检索的图片,采用已训练的卷积神经网络模型进行特征提取,得到特征向量;根据已构造的随机kd#树森林索引,进行图片特征向量的查找,得到Top#N结果。本发明相对于原始的kd#树算法更能适应高维情况,而且非常适用于分布式环境,由于每棵kd#树都可以独立进行查找,最后将结果再合并,非常符合map#reduce编程模型。
- 单位