【TensorFlow系列】【五】利用inception v3 pb模型文件做预测

2019独角兽企业重金招聘Python工程师标准>>>

本文介绍如何利用imagenet比赛上训练好的inception v3冻结的pb模型进行inference。

1.下载inception v3 pb文件。

2.导入pb到TensorFlow。

3.获取输入与预测Tensor。

4.加载图片

5.进行inference

【一】先看代码

import tensorflow as tf
import numpy as np
'''
下载训练好的pb文件
'http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz'
'''
pb_path = r"D:\TensorFlow-model\inception-2015-12-05\classify_image_graph_def.pb"
with tf.gfile.FastGFile(pb_path,'rb') as f:graph_def = tf.GraphDef()graph_def.ParseFromString(f.read())tf.import_graph_def(graph_def, name='')
with tf.Session() as session:#获取pb文件中模型的所有op，主要是为了获得input与outputprint(tf.get_default_graph().get_operations())image = "D:\TensorFlow-model\inception-2015-12-05\cropped_panda.jpg"#解码图片作为inference的输入image_data = tf.gfile.FastGFile(image, 'rb').read()softmax_tensor = session.graph.get_tensor_by_name('softmax:0')predictions = session.run(softmax_tensor,{'DecodeJpeg/contents:0': image_data})index = np.argmax(predictions,1)print(index)

结果如下：

label为169，从文件中找到169是哪个类别

以下图片中的文件，来自于上述代码链接中下载的压缩包解压后的文件。

该文件说明了label属于哪个分类

再在如下文件中查找：

是说：该图片是一直熊猫

转载于:https://my.oschina.net/u/3800567/blog/1637874

【TensorFlow系列】【五】利用inception v3 pb模型文件做预测相关推荐

查看tensorflow pb模型文件的节点信息
查看tensorflow pb模型文件的节点信息: import tensorflow as tf with tf.Session() as sess:with open('./quantized_m ...
谷歌发布TensorFlow 1.4与TensorFlow Lattice：利用先验知识提升模型准确度搜狐科技 10-12 15:29 选自：Google Research Blog 参与：李泽南、
谷歌发布TensorFlow 1.4与TensorFlow Lattice:利用先验知识提升模型准确度昨天,谷歌发布了 TensorFlow 1.4.0 先行版,将 tf.data 等功能加入了 A ...
【python脚本系列】利用mido库解析midi文件
[python脚本系列]利用mido库解析midi文件 Hallo大家好~~我是Lampard猿奋昨天写了关于Midi数字化乐器接口的文章,介绍了midi是一组代表音乐参数( ...
移动端利用OpenGL展示3D模型文件STL
移动端利用OpenGL展示3D模型文件STL 突然发现上次写博客都是一年前了,没养成分享的习惯挺郁闷的,所以分享下个人感觉好玩的东西吧.纯理工科生笔杆子不硬,写的不好,哪里有看不懂的或者写的不好的希望 ...
Retrain a tensorflow model based on Inception v3
本文在谷歌2015_CVPR Inception v3模型的基础上,结合花朵识别的具体问题重新训练该模型,以获取自己需要的tensorflow模型. 重新训练Inception v3实质是在原有模型输 ...
vc++ 调用tensorflow 2.0 的.pb模型进行神经网络预测
vc++(vs2017版本)下调用python生成的tensorflow2.0 xx.pb模型进行神经网络图片预测. 代码: #pragma once class CMyAiMode { public ...
tensorflow 回归的例子，包括保存模型和重新预测
import pathlibimport matplotlib.pyplot as plt import pandas as pd import seaborn as snsimport tensor ...
读取pb模型进行预测
本程序是解析一个tfrecord文件数据,然后调用训练好的pb模型文件去预测这些数据的类别,返回一个列表. 之前的训练程序和需要的数据到可以到这儿找:https://blog.csdn.net/mac ...
Inception V3 的 tensorflow 实现
tensorflow 官方给出的实现:models/inception_v3.py at master · tensorflow/models · GitHub 1. 模型结构首先来看 Incept ...

【TensorFlow系列】【五】利用inception v3 pb模型文件做预测

【TensorFlow系列】【五】利用inception v3 pb模型文件做预测相关推荐

最新文章

热门文章