Tensorflow服务不返回Base64编码图像

2024-04-26 21:43:35 发布

您现在位置:Python中文网/ 问答频道 /正文

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

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

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

{'predictions': [imagebytes]}

这当然会引起很多问题。我不知道我遗漏了什么,起初我认为模型服务器会返回b64编码的图像,这是意料之中的事情,因为模型服务器会自动解码请求中的b64输入。如何启用对此的支持

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

希望你的建议

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


Tags: 字符串模型图像服务器json编码字节架构