保存模型转onnx并进行推理

tensorflow保存模型转成onnx文件
参考：https://github.com/onnx/tensorflow-onnx的getting started部分
这里为了使得转化过程最简化tensorflow保存模型时使用save_model保存

from tensorflow.saved_model import simple_save
import shutil
save_path = r'/.../saved_model'
if os.path.exists(save_path):
    shutil.rmtree(save_path)
simple_save(sess, save_path, {
   x.name: x}, {
   outputs.name: outputs})

使用下列代码转化tensorflow save_model保存后的模型为onnx

python -m tf2onnx.convert --saved-model /.../saved_model/ --output /...test.onnx

当tf模型存在3d卷积时会出现下列错误