Tensorflow模型实现预测或识别单张图片


Posted in Python onJuly 19, 2019

利用Tensorflow训练好的模型,图片进行预测和识别,并输出相应的标签和预测概率。

如果想要多张图片,可以进行批次加载和预测,这里仅用单张图片进行演示。

模型文件:

Tensorflow模型实现预测或识别单张图片

预测图片:

Tensorflow模型实现预测或识别单张图片

这里直接贴代码,都有注释,应该很好理解

import tensorflow as tf
import inference
 
image_size = 128 # 输入层图片大小
 
# 模型保存的路径和文件名
MODEL_SAVE_PATH = "model/"
MODEL_NAME = "model.ckpt"
 
# 加载需要预测的图片
image_data = tf.gfile.FastGFile("./data/test/d.png", 'rb').read()
 
# 将图片格式转换成我们所需要的矩阵格式,第二个参数为1,代表1维
decode_image = tf.image.decode_png(image_data, 1)
 
# 再把数据格式转换成能运算的float32
decode_image = tf.image.convert_image_dtype(decode_image, tf.float32)
 
# 转换成指定的输入格式形状
image = tf.reshape(decode_image, [-1, image_size, image_size, 1])
 
# 定义预测结果为logit值最大的分类,这里是前向传播算法,也就是卷积层、池化层、全连接层那部分
test_logit = inference.inference(image, train=False, regularizer=None)
 
# 利用softmax来获取概率
probabilities = tf.nn.softmax(test_logit)
 
# 获取最大概率的标签位置
correct_prediction = tf.argmax(test_logit, 1)
 
# 定义Savar类
saver = tf.train.Saver()
 
with tf.Session() as sess:
  sess.run((tf.global_variables_initializer(), tf.local_variables_initializer()))
 
  # 加载检查点状态,这里会获取最新训练好的模型
  ckpt = tf.train.get_checkpoint_state(MODEL_SAVE_PATH)
  if ckpt and ckpt.model_checkpoint_path:
    # 加载模型和训练好的参数
    saver.restore(sess, ckpt.model_checkpoint_path)
    print("加载模型成功:" + ckpt.model_checkpoint_path)
 
    # 通过文件名得到模型保存时迭代的轮数.格式:model.ckpt-6000.data-00000-of-00001
    global_step = ckpt.model_checkpoint_path.split('/')[-1].split('-')[-1]
 
    # 获取预测结果
    probabilities, label = sess.run([probabilities, correct_prediction])
 
    # 获取此标签的概率
    probability = probabilities[0][label]
 
    print("After %s training step(s),validation label = %d, has %g probability" % (global_step, label, probability))
  else:
    print("模型加载失败!" + ckpt.model_checkpoint_path)

运行输出结果:

Tensorflow模型实现预测或识别单张图片

(标签为3,概率为0.984478)

标签字典:

Tensorflow模型实现预测或识别单张图片

3对应小写d,识别正确。

其他的图片的预测结果:

预测图片1:

Tensorflow模型实现预测或识别单张图片

Tensorflow模型实现预测或识别单张图片

标签字典:

Tensorflow模型实现预测或识别单张图片

图片1,识别结果为1,可能概率0.993034

识别结果还是挺好看的,不知道是不是过拟合了,还是迭代次数不够多,还需要调整调整。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python模块搜索概念介绍及模块安装方法介绍
Jun 03 Python
python密码错误三次锁定(实例讲解)
Nov 14 Python
python3实现163邮箱SMTP发送邮件
May 22 Python
Python3爬虫之urllib携带cookie爬取网页的方法
Dec 28 Python
解决python测试opencv时imread导致的错误问题
Jan 26 Python
python实现微信定时每天和女友发送消息
Apr 29 Python
如何写python的配置文件
Jun 07 Python
Python enumerate() 函数如何实现索引功能
Jun 29 Python
Windows 平台做 Python 开发的最佳组合(推荐)
Jul 27 Python
Pytorch模型迁移和迁移学习,导入部分模型参数的操作
Mar 03 Python
python 爬取京东指定商品评论并进行情感分析
May 27 Python
Python如何让字典保持有序排列
Apr 29 Python
python django下载大的csv文件实现方法分析
Jul 19 #Python
python使用flask与js进行前后台交互的例子
Jul 19 #Python
Django 模型类(models.py)的定义详解
Jul 19 #Python
Tensorflow实现神经网络拟合线性回归
Jul 19 #Python
使用TensorFlow实现简单线性回归模型
Jul 19 #Python
详解将Python程序(.py)转换为Windows可执行文件(.exe)
Jul 19 #Python
简单了解django缓存方式及配置
Jul 19 #Python
You might like
phpinfo 系统查看参数函数代码
2009/06/05 PHP
PHP CURL获取返回值的方法
2014/05/04 PHP
ThinkPHP中调用PHPExcel的实现代码
2017/04/08 PHP
php将html转为图片的实现方法
2017/05/19 PHP
javascript 网页跳转的方法
2008/12/24 Javascript
jQuery+.net实现浏览更多内容(改编php版本)
2013/03/28 Javascript
javascript中通过arguments参数伪装方法重载
2014/10/08 Javascript
jQuery实现在下拉列表选择时获取json数据的方法
2015/04/16 Javascript
原生js与jQuery实现简单的tab切换特效对比
2015/07/30 Javascript
javascript获取系统当前时间的方法
2015/11/19 Javascript
JavaScript 七大技巧(一)
2015/12/13 Javascript
7个去伪存真的JavaScript面试题
2016/01/07 Javascript
JavaScript实现异步图像上传功能
2018/07/12 Javascript
Vue中的异步组件函数实现代码
2018/07/20 Javascript
[02:44]完美大师赛主赛事淘汰赛第二日观众采访
2017/11/24 DOTA
Python实现获取网站PR及百度权重
2015/01/21 Python
python通过BF算法实现关键词匹配的方法
2015/03/13 Python
Python中的模块导入和读取键盘输入的方法
2015/10/16 Python
解决Python中list里的中文输出到html模板里的问题
2018/12/17 Python
Python二进制文件读取并转换为浮点数详解
2019/06/25 Python
python networkx 根据图的权重画图实现
2019/07/10 Python
Pycharm插件(Grep Console)自定义规则输出颜色日志的方法
2020/05/27 Python
基于Python爬虫采集天气网实时信息
2020/06/05 Python
爱奇艺VIP会员:大剧抢先看
2018/07/11 全球购物
全球性的在线鞋类品牌:Public Desire
2019/04/03 全球购物
英国No.1体育用品零售商:SportsDirect.com
2019/10/16 全球购物
高中毕业生自我鉴定范文
2013/09/26 职场文书
绝对经典成功的大学生推荐信
2013/11/08 职场文书
喷漆工的岗位职责
2014/03/17 职场文书
反邪教标语
2014/06/23 职场文书
法人代表身份证明书及授权委托书
2014/09/16 职场文书
群众路线党员个人剖析材料
2014/10/08 职场文书
手术室护士个人总结
2015/02/13 职场文书
重阳节座谈会主持词
2015/07/03 职场文书
2019初中学生入团申请书
2019/06/27 职场文书
golang生成并解析JSON
2022/04/14 Golang