项目地址
项目中首先需要安装onnx
运行:python2 weight_to_onnx.py
完成模型转换
但是遇到如下问题:
当我的onnx为最新版本1.5.0时报错:
onnx.onnx_cpp2py_export.checker.ValidationError: Node (086_upsample) has input size 1 not in range [min=2, max=2]
逛论坛发现需要把onnx降低到1.4.1
降低之后仍然出现相同错误
无奈之下直接把onnx版本降低到1.2.1:
pip2 uninstall onnx
pip2 install onnx==1.2.1
成功执行模型转化