我已经训练了一个基于Inception的网络来学习新课程。输入张量的形状为[None,2048] ...我的测试输入图像是使用像素尺寸为480 x 720的照相亭mac应用拍摄的。
我如何输入新图像以获得新图像的预测?
这是我的测试设置:
import argparse
import tensorflow as tf
import utils
from PIL import Image
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument("--pb_file", required=True, type=str, help="Frozen model file to import")
parser.add_argument("--img_path", required=True, type=str, help="Path to image for input to model")
args = parser.parse_args()
graph = utils.load_graph(args.pb_file)
for op in graph.get_operations():
print(op.name + '\t' + str(op.values()))
x = graph.get_tensor_by_name('input_ph:0')
y = graph.get_tensor_by_name('FC_layers_1/FC_4/Softmax:0')
with tf.Session(graph=graph) as sess:
feed_dict = {x: Image.open(args.img_path)}
y_out = sess.run(y, feed_dict=feed_dict)
print(y_out)
--pb_file
参数适用于最终的冻结模型
--img_path
是我要输入经过训练的模型的图像
input_ph:0
是输入占位符
我期望FC_layers_1/FC_4/Softmax:0
softmax输出