Skip to content

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

License

Notifications You must be signed in to change notification settings

FilipJergleAlmquist/onnx-to-keras

 
 

Repository files navigation

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

About

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

Resources

License

Code of conduct

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.0%
  • Other 1.0%