任何将张量流模型转换为caffe模型的代码？

Question

我的最终目标是使用TensorRT来优化我的模型以进行部署。我在tensorflow做我的实验。但TensorRT需要caffemodel格式的输入模型。

Answer 1

截至TensorRT 2.1，不支持转换TensorFlow模型。这可能会在将来得到支持。

您可能想要探索将TensorFlow模型转换为Caffe的选项，然后使用TensorRT将其转换为推理引擎。

Answer 2

新TensorRT 3增加了对TensorFlow的支持。