Tensorflow 从onnx转.pb

Posted 2023-05-15

参考技术A 首先需要安装onnx和onnx-tf这两个包，用pip安装就行

pytorch 转tensorflow注意

参考技术A 最近在作pytorch 模型转tensorflow，通过onnx 中间转换和容易，但是再转换时有一个注意事项，即如何处理batch

pytorch 模型转tensorflow： https://www.jianshu.com/p/3e5623696a8e

通过 onnx 手动修改batch 为动态值： model_onnx.graph.input[0].type.tensor_type.shape.dim[0].dim_param ='?'

这下就可以使用batch 了。

最后别忘了用transform_graph 压缩下模型大小： https://www.jianshu.com/p/d2637646cda1