基于整体轮廓的以图搜图
2021-1-16 22:59 更新
技术选型:vgg16、vgg19、resnet18、keras、numpy、pandas、opencv、hnsw、annoy 项目描述:该项目为垂直领域的以图搜图,用户上传照片,直接从数据库返回搜索相似的结果 技术负责: 1、加载pretrained model, 在我们的数据集上进行fine-tuning 2、使用fine-tuning的模型,抽取现有数据库数据的特征,存储到h5py文件中 3、当用户输入相应的图片,通过预训练的模型抽取该图像特征。 4、将抽取到的图像特征和库里已有的图像进行相似度比较,取top10进行排序,通过re-rank拿到最后结果已经结果置信度。