Skip to content

lightsalt2011/onnx_learn

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

onnx_learn

记录学习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工具类

学习笔记

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%