pytorch以图搜图通过cnn模型提取特征建立

参考链接:https://github.com/lonngxiang/image_retrieval_platform

复制这段内容后打开百度网盘手机App,操作更方便哦
pytorch以图搜图通过cnn模型提取特征建立_第1张图片
首先需要在retrieval文件下创建models文件夹,然后下载预训练参数放到models下 https://drive.google.com/open?id=1TG_Fq_UryffsmV045u4MJGaWB-MJqNgI (百度网盘链接: https://pan.baidu.com/s/1jtUKPe-q8RZqCwTdtDq5wg 提取码: 9mts ) ,static下创建thumb_images文件夹

最后运行代码:python image_retrieval.py
pytorch以图搜图通过cnn模型提取特征建立_第2张图片
上传的是这张猫搜索类似:
pytorch以图搜图通过cnn模型提取特征建立_第3张图片
pytorch以图搜图通过cnn模型提取特征建立_第4张图片

你可能感兴趣的:(深度学习)