我正在使用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模型
任何人都可以帮助您了解这个问题吗?