Tensorflow学习：使用inception-v3做图像识别

Tensorflow学习：使用inception-v3做图像识别

import tensorflow as tf
import os
import numpy as np
from PIL import Image
import matplotlib.pyplot as plt

'''数据预处理'''


class NodeLookup(object):
    def __init__(self):
        label_lookup_path = 'inception_model/imagenet_2012_challenge_label_map_proto.pbtxt'
        uid_lookup_path = 'inception_model/imagenet_synset_to_human_label_map.txt'
        self.node_lookup = self.load(label_lookup_path, uid_lookup_path)

    # 返回分类编号1-1000对应分类名称的映射关系
    def load(self, label_lookup_path, uid_lookup_path):

        # 加载分类字符串编号和对应分类名称的文件
        proto_as_ascii_lines = tf.gfile.GFile(uid_lookup_path).readlines()
        uid_to_human = {
   }
        # 一行一行读取数据
        for line in proto_as_ascii_lines:
            # 去掉换行符
            line = line.strip('\n')
            # 按'\t'分割
            parsed_items = line.split('\t')
            # 获取分类编号
            uid = parsed_items[0]
            # 获取分类名称
            human_string = parsed_items[1]
            # 保存编号和分类名称的映射关系
            uid_to_human[uid] = human_string

        # 加载分类字符串编号和对应分类编号1-1000的文件
        proto_as_ascii = tf.gfile.GFile