MNIST手写数字做图像检索(Tensorflow)

参考:https://blog.csdn.net/oolyjoo38/article/details/71703162?locationNum=4&fps=1
郑泽宇著 《TensorFlow实战Google深度学习框架》
黄文坚著 《TensorFlow实战》

温馨提示:需要这两本书的同学可以去我的网盘下载
链接:https://pan.baidu.com/s/1fFl0j2iLiZUnD52ob24OIA
提取码:0civ

检索的思路有两种方式:
第一种是基于分类的检索,简单的来讲就是先做分类,然后输入一张测试图像,判别属于某一类,在把测试集中所有属于这一类的保存即可;
第二种是基于特征的检索,简单的讲就是全连接层会输出维度为1024的特征,输入一张测试图片提取其特征,再将库里所有图像的特征都提取出来,进行特征距离的计算,然后根据实际情况设定一个阈值,低于这个阈值的图像保存即可;
我才用第二种方法。

环境:ubuntu16.04; python 3.5.2; tensorflow;
源代码下载:

稍后放上代码链接和训练结果!

猜你喜欢

转载自blog.csdn.net/jingjingliang1995/article/details/89705599