tensorflow - 任何将 tensorflow 模型转换为 caffe 模型的代码？

Question

我的最终目标是使用 TensorRT 来优化我的部署模型。我在张量流中做我的实验。但是 TensorRT 需要 caffemodel 格式的输入模型。

Answer 1

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

Answer 2

从 TensorRT 2.1 开始，不支持转换 TensorFlow 模型。这可能会在未来得到支持。

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