GitPlanet

Cheap and reliable Node.js hosting starts at $3/month, and $1/month static HTML hosting

Created with love in Canada, visit hostnodejs.com today

Feel like to post an Ad? Learn Details
All Projects → AxisCommunications → onnx-to-keras

AxisCommunications / onnx-to-keras

Licence: MIT license
Convert onnx models exported from pytorch to tensorflow keras models with focus on performace and highleve compatibility.

Programming Languages

python
139335 projects - #7 most used programming language

onnx2keras

This is a tool for converting onnx models (as exported by for example pytorch) into tensorflow keras models. It focuses on inference performance and what we call high-level-compatibility rather than completeness. That is, it will not be able to convert every onnx model, but the models it can convert it will convert in a nice way. With high-level-compatibility we mean that the converted models produced are constructed using the high-level keras API and should be similar to how the model would have been implemented in keras if it was implemented by hand.

Usage

    python3 onnx2keras.py <infile.onnx> [<outfile.h5>]

Run tests

    make
Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].