ModECI / MDF

This repository contains the source for the MDF specification and Python API
https://mdf.readthedocs.io
Apache License 2.0
36 stars 69 forks source link

Keras to MDF #491

Open Onoyiza opened 1 year ago

Onoyiza commented 1 year ago

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:

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 commented 1 year ago

Thanks for adding details on this @Onoyiza!