Tensorflow Serving不返回Base64编码图像

时间:2020-04-23 16:45:51

标签: python rest tensorflow base64 tensorflow-serving

我使用Tensorflow Serving部署了基于Conv的神经网络架构。我正在查询Rest API,正如预期的那样,我在json正文上发送了Base64编码的图像:

{'instances' : [{'b64': b64image}]}

一切正常,但Tensorflow Model Server的json响应不会返回其他b64编码的图像,而是原始jpg文件字节:

{'predictions': [imagebytes]}

这当然会引起很多问题。我不知道我缺少什么,起初我以为Model Server返回b64编码的图像,这是预料之中的,因为Model Server自动解码了请求中的b64输入。如何为此提供支持?

现在我能想到的解决方案是包装我的模型以返回b64编码的字符串(实际上我的模型已经包装好以用作输入原始图像字节并输出原始图像字节)

希望您的建议!

PD:我决定修改导出的模型以输出b64编码的图像字符串,这现在解决了我的问题,但是可以肯定的是,这必须有另一种方法,TF Serving支持输入的Base64解码是不合逻辑的,但是不会以相同的方式编码输出。

0 个答案:

没有答案