从Keras模型转换成tensorrt的方法

1 流程 对于目前找到的最有效的方式是将Keras模型保存为savedmodel格式,也就是说一个只含有计算图的.pb,一个variables文件和一个json合并的一个文件夹。tf-trt加速的话会还给你一个优化过后的.pb和variable,并不是我们希望使用的engine形式。 最稳妥的流程为:savedmodel–>onnx–>engine。 下载tf2onnx,使用如下命令: pytho
相关文章
相关标签/搜索