上传者: 38747211
|
上传时间: 2021-06-21 10:40:16
|
文件大小: 65KB
|
文件类型: PDF
利用Tensorflow训练好的模型,图片进行预测和识别,并输出相应的标签和预测概率。
如果想要多张图片,可以进行批次加载和预测,这里仅用单张图片进行演示。
模型文件:
预测图片:
这里直接贴代码,都有注释,应该很好理解
import tensorflow as tf
import inference
image_size = 128 # 输入层图片大小
# 模型保存的路径和文件名
MODEL_SAVE_PATH = model/
MODEL_NAME = model.ckpt
# 加载需要预测的图片
image_data = tf.gfile.FastGFile(./data