张量流服务-批量重塑错误

时间:2018-09-26 01:41:53

标签: python-3.x tensorflow tensorflow-serving batching

我正在使用inception_client.py通过张量流服务执行预测,并且效果很好。但是当我尝试按请求批处理图像并执行时,tensorflow服务API会引发此异常:

  

AbortionError(code = StatusCode.INVALID_ARGUMENT,details =“要重塑的输入是具有20480个值的张量,但请求的形状为2048        [[Node:pool_3 / _reshape = Reshape [T = DT_FLOAT,Tshape = DT_INT32,_output_shapes = [[1,2048]],_device =“ / job:localhost /副本:0 /任务:0 /设备:CPU:0” ](pool_3,pool_3 / _reshape / shape)]]“)

作为预处理,我阅读了两张图像,转换为张量,进行规范化并创建一个列表,以在make_tensor_proto上使用:

proto = tf.contrib.util.make_tensor_proto(images, shape=tf.shape(images).numpy().tolist())
request.inputs['image'].CopyFrom(proto)
result = stub.Predict(request, 120.0)

我已经使用retrain.py创建了该模型,并重新训练了InceptionV3模型

任何人都可以帮助您了解这个问题吗?

0 个答案:

没有答案
相关问题