Skip to content

Keras to MDF #491

@Onoyiza

Description

@Onoyiza

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions