利用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)
运行输出结果:
(标签为3,概率为0.984478)
标签字典:
3对应小写d,识别正确。
其他的图片的预测结果:
预测图片1:
标签字典:
图片1,识别结果为1,可能概率0.993034
识别结果还是挺好看的,不知道是不是过拟合了,还是迭代次数不够多,还需要调整调整。
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。
华山资源网 Design By www.eoogi.com
广告合作:本站广告合作请联系QQ:858582 申请时备注:广告合作(否则不回)
免责声明:本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除!
免责声明:本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除!
华山资源网 Design By www.eoogi.com
暂无评论...
稳了!魔兽国服回归的3条重磅消息!官宣时间再确认!
昨天有一位朋友在大神群里分享,自己亚服账号被封号之后居然弹出了国服的封号信息对话框。
这里面让他访问的是一个国服的战网网址,com.cn和后面的zh都非常明白地表明这就是国服战网。
而他在复制这个网址并且进行登录之后,确实是网易的网址,也就是我们熟悉的停服之后国服发布的暴雪游戏产品运营到期开放退款的说明。这是一件比较奇怪的事情,因为以前都没有出现这样的情况,现在突然提示跳转到国服战网的网址,是不是说明了简体中文客户端已经开始进行更新了呢?
更新日志
2024年09月24日
2024年09月24日
- 国风大师纯音系列《古筝仙子 常静演奏 十指弹一》1CD[FLAC][919MB]
- DasKapital-OneMustHaveChaosInsidetoGiveBirthtoaDancingStar(2024)[24-44.1]WAV
- 王菲.1999-《只爱陌生人》日本东芝版[低速原抓WAV+CUE]
- 张小英1987-金奖金曲第一集[新加坡版][WAV+CUE]
- 鹿晗.2015-重启【海蝶】【FLAC分轨】
- 张芸京.2012-小女孩【金牌大风】【WAV+CUE】
- 余天.2002-雄狮魅力金曲2CD(南方金点系列)【南方】【WAV+CUE】
- 银霞.1982-《结·SAYYESMYBOY》台湾复刻版[WAV+CUE]
- 梁汉文1993-火热动感[引进版][WAV+CUE]
- 梁汉文2008-EDMONDHITS48[香港]3CD[WAV+CUE]
- 国风大师纯音系列《童丽-[渭城曲]》1CD[WAV分轨][593.5MB]
- 国风大师纯音系列《许菱子《烧筝2》古筝曲集》1CD[WAV分轨][1.3GB]
- 王子异《1996 Part.2》[320K/MP3][18.29MB]
- 群星.1994-宝丽金最倾情3CD【宝丽金】【WAV+CUE】
- 乐俊亨.2024-同泪【CharioticMusic】【FLAC分轨】