无法为形状为'(?,2048)'的张量u'input_ph:0'输入形状(480、720、3)的值

时间:2018-08-12 19:08:50

标签: python tensorflow machine-learning computer-vision

我已经训练了一个基于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输出

0 个答案:

没有答案