將tensorflow模型轉換爲caffe模型的任何代碼？

Question

我的最終目標是使用TensorRT優化我的部署模型。我正在做tensorflow中的實驗。但是TensorRT需要caffemodel格式的輸入模型。

Answer 1

從TensorRT 2.1開始，就不支持轉換TensorFlow模型。這可能會在未來得到支持。

您可能想要探索將TensorFlow模型轉換爲Caffe的選項，然後使用TensorRT將其轉換爲推理引擎。

Answer 2

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