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

Question

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

Answer 1

从 TensorRT 2.1 开始，不支持转换 TensorFlow 模型。将来可能会支持此功能。

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

Answer 2

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