使用TensorFlow自带的实例程序完成分类任务,现将整个实现过程记录如下:
左图为输入图,右图为输出结果在label文件内的位置(位置-1=索引)
工程文件简介
1. 根据在Git上下载的最新tensorflow代码,找到label_image工程,查看Readme文件,下载程序运行所需的文件:
2. 下载Graph文件,地址如下:
3. 下载后的文件如下:
4. 解压,将labels、.pb文件放到data目录下,如下所示:
运行过程
在Anaconda Prompt内,进入工程目录下,输入命令
python label_image.py
如下:
结果
运行结果如下,显示分数最高的前5个: