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.

Sign up for GitHub

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

Jump to bottom

Add export to ONNX #161

Merged

chandrikadeb7 merged 1 commit into chandrikadeb7:master from SthPhoenix:add_onnx_export

Jan 1, 2022

Contributor

SthPhoenix commented Dec 31, 2021

Describe the changes you've made

Add script model2onnx.py to export keras model to ONNX

Describe if there is any unusual behaviour of your code(Write `NA` if there isn't)

NA

Additional context (OPTIONAL)

With proposed script it's possible to export model to ONNX which can be used in ONNXRuntime or TensorRT, both were tested.

Test plan (OPTIONAL)

Checklist

My code follows the code style of this project.
I have performed a self-review of my own code.
My change requires a change to the documentation.
I have updated the documentation accordingly.
I have commented my code, particularly in hard-to-understand areas.
My changes generate no new warnings.
I have added tests that prove my fix is effective or that my feature works.
The title of my pull request is a short description of the requested changes.


          Add export to ONNX

c90b9c1

chandrikadeb7 merged commit 4406861 into chandrikadeb7:master

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment