tensorflow卷积网络(TensorFlow卷积神经网络之使用训练好的模型识别猫狗图片)
类别:脚本大全 浏览量:2835
时间:2021-11-08 16:38:57 tensorflow卷积网络
TensorFlow卷积神经网络之使用训练好的模型识别猫狗图片本文是Python通过TensorFlow卷积神经网络实现猫狗识别的姊妹篇,是加载上一篇训练好的模型,进行猫狗识别
本文逻辑:
- 我从网上下载了十几张猫和狗的图片,用于检验我们训练好的模型。
- 处理我们下载的图片
- 加载模型
- 将图片输入模型进行检验
代码如下:
|
#coding=utf-8 import tensorflow as tf from PIL import Image import matplotlib.pyplot as plt import input_data import numpy as np import model import os #从指定目录中选取一张图片 def get_one_image(train): files = os.listdir(train) n = len (files) ind = np.random.randint( 0 ,n) img_dir = os.path.join(train,files[ind]) image = Image. open (img_dir) plt.imshow(image) plt.show() image = image.resize([ 208 , 208 ]) image = np.array(image) return image def evaluate_one_image(): #存放的是我从百度下载的猫狗图片路径 train = '/Users/yangyibo/GitWork/pythonLean/AI/猫狗识别/testImg/' image_array = get_one_image(train) with tf.Graph().as_default(): BATCH_SIZE = 1 # 因为只读取一副图片 所以batch 设置为1 N_CLASSES = 2 # 2个输出神经元,[1,0] 或者 [0,1]猫和狗的概率 # 转化图片格式 image = tf.cast(image_array, tf.float32) # 图片标准化 image = tf.image.per_image_standardization(image) # 图片原来是三维的 [208, 208, 3] 重新定义图片形状 改为一个4D 四维的 tensor image = tf.reshape(image, [ 1 , 208 , 208 , 3 ]) logit = model.inference(image, BATCH_SIZE, N_CLASSES) # 因为 inference 的返回没有用激活函数,所以在这里对结果用softmax 激活 logit = tf.nn.softmax(logit) # 用最原始的输入数据的方式向模型输入数据 placeholder x = tf.placeholder(tf.float32, shape = [ 208 , 208 , 3 ]) # 我门存放模型的路径 logs_train_dir = '/Users/yangyibo/GitWork/pythonLean/AI/猫狗识别/saveNet/' # 定义saver saver = tf.train.Saver() with tf.Session() as sess: print ( "从指定的路径中加载模型。。。。" ) # 将模型加载到sess 中 ckpt = tf.train.get_checkpoint_state(logs_train_dir) if ckpt and ckpt.model_checkpoint_path: global_step = ckpt.model_checkpoint_path.split( '/' )[ - 1 ].split( '-' )[ - 1 ] saver.restore(sess, ckpt.model_checkpoint_path) print ( '模型加载成功, 训练的步数为 %s' % global_step) else : print ( '模型加载失败,,,文件没有找到' ) # 将图片输入到模型计算 prediction = sess.run(logit, feed_dict = {x: image_array}) # 获取输出结果中最大概率的索引 max_index = np.argmax(prediction) if max_index = = 0 : print ( '猫的概率 %.6f' % prediction[:, 0 ]) else : print ( '狗的概率 %.6f' % prediction[:, 1 ]) # 测试 evaluate_one_image() |
/Users/yangyibo/GitWork/pythonLean/AI/猫狗识别/testImg/ 存放的是我从百度下载的猫狗图片
执行结果:
因为从testimg 中选取图片是随机的,所以每次执行的结果不同
从指定的路径中加载模型。。。。
模型加载成功, 训练的步数为 11999
狗的概率 0.964047
[Finished in 6.8s]
代码地址:https://github.com/527515025/My-TensorFlow-tutorials/blob/master/猫狗识别/evaluateCatOrDog.py
欢迎star。
总结
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,谢谢大家对开心学习网的支持。如果你想了解更多相关内容请查看下面相关链接
原文链接:https://blog.csdn.net/u012373815/article/details/79222121
您可能感兴趣
- tensorflow卷积网络(TensorFlow卷积神经网络之使用训练好的模型识别猫狗图片)
- 卷积神经网络python实现(Python通过TensorFlow卷积神经网络实现猫狗识别)
- 谢广坤,你这么欺负谢腾飞,良心不会痛吗(你这么欺负谢腾飞)
- 乡村爱情15 宋晓峰怀疑自己孩子,腾飞与姜奶奶亲子鉴定出结果(宋晓峰怀疑自己孩子)
- 《乡村爱情13》开播,新版刘能以假乱真,编剧思维进入瓶颈(新版刘能以假乱真)
- 当年的 白洋淀战神 练肌肉 嘎子哥也成为行走的荷尔蒙(当年的白洋淀战神)
- 肌肉小子陈康, 亚洲巨兽 黄哲勋,哪个才是你的菜(肌肉小子陈康亚洲巨兽)
- 新闻周刊 青岛网红 赵厂长 编段子一箩筐输出快乐,陪父亲十二载勇斗病魔(新闻周刊青岛网红)
热门推荐
- vue 父组件向子组件传值几种方法(Vue中父组件向子组件传递数据的几种方法)
- python中for循环计算1-100(python 用for循环实现1~n求和的实例)
- scrollWidth,clientWidth,offsetWidth的区别
- mysql大量数据怎么处理(MySQL删除和插入数据很慢的问题解决)
- 云服务器租用怎么收费(云服务器租用注意事项有哪些?)
- mysql存储json的方式(MySQL中查询json格式的字段实例详解)
- python 正则表达式语法大全(python re库的正则表达式入门学习教程)
- 前端单行文本垂直居中(移动端的text-overflow多行文本溢出显示省略号…)
- dedecms 内容页模板(dedecms列表页与详情页调用tag标签的方法)
- python中字典的常用操作(11个Python3字典内置方法大全与示例汇总)