Tensorflow服务与对象检测

时间:2018-07-24 21:39:06

标签: tensorflow object-detection tensorflow-serving

我正在尝试基于“对象检测”示例部署模型进行一些测试,但出现此错误:

“预计arg [0]为uint8,但提供了浮点数”

在这种情况下,我正在使用它来加载我的数据:


request.inputs ['inputs']。CopyFrom(         tf.contrib.util.make_tensor_proto({FLAGS.input_image}))


其中FLAGS.input_image是我的图像数据(以字节为单位)。

我当时想也许应该将图像字节转换为此输入可以理解的格式,但是我还没有找到。

该如何解决此问题?

谢谢!!!!

1 个答案:

答案 0 :(得分:1)

要将图像转换为字节,请在客户端代码(python)中使用以下内容

with open(FLAGS.image, 'rb') as f:
     data = f.read()

还请按照以下说明https://github.com/tensorflow/serving/blob/master/tensorflow_serving/example/inception_client.py

查找示例客户端(用于python中的初始模型)