MDF icon indicating copy to clipboard operation
MDF copied to clipboard

Keras to MDF

Open Onoyiza opened this issue 2 years ago • 1 comments

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

Onoyiza avatar Sep 28 '23 12:09 Onoyiza

Thanks for adding details on this @Onoyiza!

pgleeson avatar Sep 28 '23 18:09 pgleeson