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/目录下

详细参考

分类:

技术点:

相关文章:

  • 2021-05-01
  • 2022-02-10
  • 2021-08-02
  • 2021-12-21
  • 2021-07-23
  • 2021-10-03
  • 2021-12-15
  • 2022-12-23
猜你喜欢
  • 2022-12-23
  • 2021-08-12
  • 2021-09-02
  • 2021-11-20
  • 2022-02-12
  • 2021-10-19
  • 2021-12-29
相关资源
相似解决方案