kikokiko

相似图像检测

VGGNet特征提取

利用VGGnet的预训练模型来实现图像的检索,先用预训练模型来抽取图片的特征,然后把待检索的图像和数据库中的所有图像进行匹配,找出相似度最高的

在jupyter notebook上实现

文件路径设置:
root|____ code
       |____ images|____ img_class_1
                           |____ img_class_2
                           |____ img_class_3
                           |.... .....
                           |____ img_class_n
       |____models
       |____queryimg

  • root: 根目录
  • images: 存放各类别的图片文件夹
  • img_class_i: 存放相应类别的图片
  • database: 用于存放数据
  • queryimg: 存放待检索图片

Step 1. 构造特征提取器

这里用了Keras的应用模块(Keras.applications)提供的带有预训练权值的模型

初始化一个模型的时候,会自动下载权重到~/.keras/models/目录下

详细参考

分类:

技术点:

相关文章: