1. tf.train.Saver(）

• tf.train.Saver()是一个类，提供了变量、模型(也称图Graph)的保存和恢复模型方法。

• TensorFlow是通过构造Graph的方式进行深度学习，任何操作(如卷积、池化等)都需要operator，保存和恢复操作也不例外。

• 在tf.train.Saver()类初始化时，用于保存和恢复的save和restore operator会被加入Graph。所以，下列类初始化操作应在搭建Graph时完成。

saver = tf.train.Saver()

TensorFlow的保存和恢复分为两种：

• 保存和恢复变量

• 保存和恢复模型

saver.save()保存模型

#举例：
保存一个训练好的手写数据集识别模型
保存在当前路径的net文件夹中

      import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data#载入数据集mnist = input_data.read_data_sets('MNIST_data', one_hot=True)#每个批次100张照片batch_size = 100#计算一个需要多少个批次n_batch = mnist.train.num_examples // batch_size#定义两个placeholderx = tf.placeholder(tf.float32, [None, 784])y = tf.placeholder(tf.float32, [None, 10])#创建一个简单的神经网络，输入层784个神经元，输出层10个神经元W = tf.Variable(tf.zeros([784, 10]))b = tf.Variable(tf.zeros([10]))prediction = tf.nn.softmax(tf.matmul(x, W) + b)#代价函数loss = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(labels=y, logits=prediction))#使用梯度下降法train_step = tf.train.GradientDescentOptimizer(0.2).minimize(loss)#初始化变量init = tf.global_variables_initializer()#结果存放在一个布尔型列表中correct_prediction = tf.equal(tf.argmax(y, 1), tf.argmax(prediction, 1))accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))saver = tf.train.Saver()with tf.Session() as sess:    sess.run(init)    for epoch in range(11):        for batch in range(n_batch):            batch_xs, batch_ys = mnist.train.next_batch(batch_size)            sess.run(train_step, feed_dict={x:batch_xs, y:batch_ys})        acc = sess.run(accuracy, feed_dict={x:mnist.test.images, y:mnist.test.labels})        print('Iter = ' + str(epoch) +', Testing Accuracy = ' + str(acc))    #保存模型    saver.save(sess, 'net/my_net.ckpt')

View Code

#保存路径中的文件为：    checkpoint：保存当前网络状态的文件
    my_net.ckpt.data-00000-of-00001
    my_net.ckpt.index
    my_net.ckpt.meta：保存Graph结构的文件

#关于函数saver.save()，常用的参数就是前三个:save(
    sess,  # 必需参数，Session对象
    save_path,  # 必需参数，存储路径
    global_step=None,  # 可以是Tensor, Tensor name, 整型数
    latest_filename=None,  # 协议缓冲文件名，默认为'checkpoint'，不用管
    meta_graph_suffix='meta',  # 图文件的后缀，默认为'.meta'，不用管
    write_meta_graph=True,  # 是否保存Graph
    write_state=True,  # 建议选择默认值True
    strip_default_attrs=False  # 是否跳过具有默认值的节点

saver.restore()加载已经训练好的模型

#举例：
通过加载刚才保存的训练好的手写数据集识别模型进行手写数据集的识别

 1 import os 2 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' 3 import tensorflow as tf 4 from tensorflow.examples.tutorials.mnist import input_data 5  6 mnist = input_data.read_data_sets('MNIST_data', one_hot=True) 7 batch_size = 100 8 n_batch = mnist.train.num_examples // batch_size 9 10 x = tf.placeholder(tf.float32, [None, 784])11 y = tf.placeholder(tf.float32, [None, 10])12 13 W = tf.Variable(tf.zeros([784, 10]))14 b = tf.Variable(tf.zeros([10]))15 prediction = tf.nn.softmax(tf.matmul(x, W) + b)16 17 loss = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(labels=y, logits=prediction))18 train_step = tf.train.GradientDescentOptimizer(0.2).minimize(loss)19 20 init = tf.global_variables_initializer()21 22 correct_prediction = tf.equal(tf.argmax(y, 1), tf.argmax(prediction, 1))23 accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))24 25 saver = tf.train.Saver()26 27 with tf.Session() as sess:28     sess.run(init)29     print(sess.run(accuracy, feed_dict={x:mnist.test.images, y:mnist.test.labels}))30     saver.restore(sess, 'net/my_net.ckpt')31     print(sess.run(accuracy, feed_dict={x:mnist.test.images, y:mnist.test.labels}))

View Code

#执行结果：
    0.098
0.9178#直接得到的准确率相当低，通过加载训练好的模型，识别准确率大大提升。

2. 下载google图像识别网络inception-v3并查看结构

模型背景：
　　Inception(v3) 模型是Google 训练好的最新一个图像识别模型，我们可以利用它来对我们的图像进行识别。

下载地址：
　　https://storage.googleapis.com/download.tensorflow.org/models/inception_dec_2015.zip

文件描述：

• classify_image_graph_def.pb 文件就是训练好的Inception-v3模型。

• imagenet_synset_to_human_label_map.txt是类别文件，包含人类标签和uid之间的映射的文件。

• imagenet_2012_challenge_label_map_proto.pbtxt是包含类号和uid之间的映射的文件。

代码实现

 1 import os 2 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' 3 import tensorflow as tf 4 import tarfile 5 import requests 6  7 #inception模型下载地址 8 inception_pretrain_model_url = 'http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz' 9 10 #inception模型存放地址11 inception_pretrain_model_dir = 'inception_model'12 if not os.path.exists(inception_pretrain_model_dir):13     os.makedirs(inception_pretrain_model_dir)14 #获取文件名，以及文件路径15 filename = inception_pretrain_model_url.split('/')[-1]16 filepath = os.path.join(inception_pretrain_model_dir, filename)17 18 #下载模型19 if not os.path.exists(filepath):20     print('download: ', filename)21     r = requests.get(inception_pretrain_model_url, stream=True)22     with open(filepath, 'wb') as f:23         for chunk in r.iter_content(chunk_size=1024):24             if chunk:25                 f.write(chunk)26     print('finish: ', filename)27 #解压文件28 tarfile.open(filepath, 'r:gz').extractall(inception_pretrain_model_dir)29 30 #模型结构存放文件31 log_dir = 'inception_log'32 if not os.path.exists(log_dir):33     os.makedirs(log_dir)34 35 #classify_image_graph_def.pb为google训练好的模型36 inception_graph_def_file = os.path.join(inception_pretrain_model_dir, 'classify_image_graph_def.pb')37 with tf.Session() as sess:38     #创建一个图来存放google训练好的模型39     with tf.gfile.FastGFile(inception_graph_def_file, 'rb') as f:40         graph_def = tf.GraphDef()41         graph_def.ParseFromString(f.read())42         tf.import_graph_def(graph_def, name='')43     #保存图的结构44     writer = tf.summary.FileWriter(log_dir, sess.graph)45     writer.close()

View Code

#在下载过程中，下的特别慢，不知道是网络原因还是什么#程序总卡着不动#所以我就手动下载压缩包并进行解压

下载结果

3. 使用inception-v3做各种图像的识别

#代码实现：

 1 import os 2 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' 3 import tensorflow as tf 4 import numpy as np 5 import re 6 from PIL import Image 7 import matplotlib.pyplot as plt 8  9 #这部分是对标签号和类别号文件进行一个预处理10 11 class NodeLookup(object):12     def __init__(self):13         label_lookup_path = 'inception_model/imagenet_2012_challenge_label_map_proto.pbtxt'14         uid_lookup_path = 'inception_model/imagenet_synset_to_human_label_map.txt'15         self.node_lookup = self.load(label_lookup_path, uid_lookup_path)16     def load(self, label_lookup_path, uid_lookup_path):17         #加载分类字符串n********对应分类名称的文件18         proto_as_ascii_lines = tf.gfile.GFile(uid_lookup_path).readlines()19         uid_to_human={}20         #一行一行读取数据21         for line in proto_as_ascii_lines:22             #去掉换行符23             line = line.strip('\n')24             #按照‘\t’进行分割25             parsed_items = line.split('\t')26             #获取分类编号27             uid = parsed_items[0]28             #获取分类名称29             human_string = parsed_items[1]30             #保存编号字符串n********与分类名称的映射关系31             uid_to_human[uid] = human_string32 33         #加载分类字符串n********对应分类编号1-1000的文件34         proto_as_ascii = tf.gfile.GFile(label_lookup_path).readlines()35         node_id_to_uid = {}36         for line in proto_as_ascii:37             if line.startswith('  target_class:'):38                 #获取分类编号1-100039                 target_class = int(line.split(': ')[1])40             if line.startswith('  target_class_string:'):41                 #获取编号字符串nn********42                 target_class_string = line.split(': ')[1]43                 # 保存分类编号1-1000与编号字符串n********映射关系44                 node_id_to_uid[target_class] = target_class_string[1:-2]45         # 建立分类编号1-1000对应分类名称的映射关系46         node_id_to_name = {}47         for key, val in node_id_to_uid.items():48             #获取分类名称49             name = uid_to_human[val]50             # 建立分类编号1-1000到分类名称的映射关系51             node_id_to_name[key] = name52         return node_id_to_name53     # 传入分类编号1-1000返回分类名称54     def id_to_string(self, node_id):55         if node_id not in self.node_lookup:56             return ''57         return self.node_lookup[node_id]58 59 #创建一个图来存放google训练好的模型60 61 with tf.gfile.FastGFile('inception_model/classify_image_graph_def.pb', 'rb') as f:62     graph_def = tf.GraphDef()63     graph_def.ParseFromString(f.read())64     tf.import_graph_def(graph_def, name='')65 66 with tf.Session() as sess:67     softmax_tensor = sess.graph.get_tensor_by_name('softmax:0')68     #遍历目录69     for root, dirs, files in os.walk('images/'):70         for file in files:71             #载入图片72             image_data = tf.gfile.FastGFile(os.path.join(root, file), 'rb').read()73             predictions = sess.run(softmax_tensor, {'DecodeJpeg/contents:0': image_data})#图片格式是jpg格式74             predictions = np.squeeze(predictions)#把结果转为1维数据75 76             #打印图片路径及名称77             image_path = os.path.join(root, file)78             print(image_path)79 80             # 显示图片81             img = Image.open(image_path)82             plt.imshow(img)83             plt.axis('off')84             plt.show()85 86             #排序87             top_k = predictions.argsort()[-5:][::-1]88             node_lookup = NodeLookup()89             for node_id in top_k:90                 # 获取分类名称91                 human_string = node_lookup.id_to_string(node_id)92                 # 获取该分类的置信度93                 score = predictions[node_id]94                 print('%s(score = %.5f)' % (human_string, score))95             print()

View Code

#执行结果：

images/1.jpg
giant panda, panda, panda bear, coon bear, Ailuropoda melanoleuca(score = 0.87265)
badger(score = 0.00260)
lesser panda, red panda, panda, bear cat, cat bear, Ailurus fulgens(score = 0.00205)
brown bear, bruin, Ursus arctos(score = 0.00102)
ice bear, polar bear, Ursus Maritimus, Thalarctos maritimus(score = 0.00099)

images/2.jpg
French bulldog(score = 0.94474)
bull mastiff(score = 0.00559)
pug, pug-dog(score = 0.00352)
Staffordshire bullterrier, Staffordshire bull terrier(score = 0.00165)
boxer(score = 0.00116)

images/3.jpg
zebra(score = 0.94011)
tiger, Panthera tigris(score = 0.00080)
pencil box, pencil case(score = 0.00066)
hartebeest(score = 0.00059)
tiger cat(score = 0.00042)

images/4.jpg
hare(score = 0.87019)
wood rabbit, cottontail, cottontail rabbit(score = 0.04802)
Angora, Angora rabbit(score = 0.00612)
wallaby, brush kangaroo(score = 0.00181)
fox squirrel, eastern fox squirrel, Sciurus niger(score = 0.00056)

images/5.jpg
fox squirrel, eastern fox squirrel, Sciurus niger(score = 0.95047)
marmot(score = 0.00265)
mongoose(score = 0.00217)
weasel(score = 0.00201)
mink(score = 0.00199)

原文出处：https://www.cnblogs.com/guoruxin/p/10238018.html