如何在从.pb文件恢复的模型输入层之前添加图层?

时间:2017-12-21 08:52:56

标签: tensorflow tensorflow-serving

我有一个.pb模型文件,我可以通过get_tensor_by_name获取输入张量和输出张量,然后使用输入和输出张量创建一个保存的模型,然后可以在tensorflow模型服务器中提供。但目前,输入张量是3D数组格式的图像,我想再添加一层来解码jpeg编码的字符串,并将结果输入到当前的输入张量中,有关如何做到这一点的任何建议吗?

1 个答案:

答案 0 :(得分:0)

您可以在此处 https://github.com/tensorflow/models/blob/master/research/slim/export_inference_graph.py#L114-L116编辑文件export_inference_graph来添加这些图层。

您可以在占位符代码之后导出推理图之前添加所需的任何预处理张量评估。