参考:pytorch模型转tensorflow的pb文件 - 知乎
pip install tensorflow-gpu==2.2.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip uninstall onnx-tf==1.6
tensorflow2.2.0+onnx-tf1.6生成单个pb文件,
参考:pytorch模型转tensorflow的pb文件 - 知乎
pip install tensorflow-gpu==2.2.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip uninstall onnx-tf==1.6
tensorflow2.2.0+onnx-tf1.6生成单个pb文件,