-
Notifications
You must be signed in to change notification settings - Fork 68
Description
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