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

Keras to MDF #491

Open
Onoyiza opened this issue Sep 28, 2023 · 1 comment
Open

Keras to MDF #491

Onoyiza opened this issue Sep 28, 2023 · 1 comment

Comments

@Onoyiza
Copy link
Contributor

Onoyiza commented Sep 28, 2023

The current Keras to MDF conversion pipeline only works when the Keras model contains an Input Layer and/ Flatten Layers and / Dense Layers. It uses a function that checks the types of layers in the Keras model and maps out the equivalent node representation in an MDF graph and model. Frequently used layers like Dropout, BatchNormalization for any network and Conv2D, Conv3D, and Maxpooling for convolutional networks do not currently work using the pipeline.

Solutions that are currently being considered:

  • Get a working equivalent node representation of the layers in MDF using Onnx operators and functions
  • Use the tf2onnx tool for the conversion (Has support for commonly used layers but not all layers)

Resources to foster solution:
Documentation on Onnx Operators
Supported Operations for Tensorflow/Keras to Onnx
Common Conversion Issues Encoutered
Sample Tutorial using Resnet model from Keras converted to Onnx
Repo for Tensorflow to Onnx Conversion

@pgleeson
Copy link
Member

Thanks for adding details on this @Onoyiza!

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

No branches or pull requests

2 participants