Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add onnx export tool #136

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

add onnx export tool #136

wants to merge 1 commit into from

Conversation

PurpleSky-NS
Copy link

@PurpleSky-NS PurpleSky-NS commented Dec 4, 2024

添加了导出onnx的工具,基本用法如下:
python tools/export_onnx.py <config> --checkpoint <checkpoint>
默认导出尺寸为224,也可以修改:
python tools/export_onnx.py <config> --checkpoint <checkpoint> --shape 256 256
也可以修改opset:
python tools/export_onnx.py <config> --checkpoint <checkpoint> --optset 16
onnx生成到checkpoint的同级目录下,并改后缀名为onnx

需要提前安装相关库:
pip install onnx onnxsim

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant