记录学习ONNX的一些资料以及提供一些实践经验。
- numpy 必选
- onnx 必选
- onnxruntime 可选,如果脚本里面使用到了onnxruntime则需要安装
- pytorch 可选,如果脚本里面使用到了pytorch则需要安装
- convert2onnx
- pytorch2onnx_resize.py 通过Pytorch导出ONNX模型,Reshape操作
- onnxapi
- creat_onnx_example.py 创建一个onnx模型例子
- tools 维护一个工具类,方便修改ONNX模型来解决ONNX版本迭代以及框架之间对OP定义的不兼容问题
- test_tool.py 测试ONNX工具类