從Keras模型轉換成tensorrt的方法

1 流程 對於目前找到的最有效的方式是將Keras模型保存爲savedmodel格式，也就是說一個只含有計算圖的.pb，一個variables文件和一個json合併的一個文件夾。tf-trt加速的話會還給你一個優化過後的.pb和variable，並不是我們希望使用的engine形式。 最穩妥的流程爲：savedmodel–>onnx–>engine。 下載tf2onnx，使用如下命令： pytho

>>阅读原文<<