Awesome

AIfES-Converter

This is a convert tool to create AIfES models for direct use in the Arduino IDE or other IDEs. It can read Feed Forward Neural Networks (FFNN) models from Keras and PyTorch and converts them to AIfES models, which are exported in header files. Those header files can than be added to your Project in any IDE and can be used there.

Quick Start

Install the converter:

pip install AIfES-Converter

IMPORTANT: For a detailed description of the installation see the documentation

Convert a Keras model, e.g.:

from aifes import keras2aifes

keras2aifes.convert_to_fnn_f32_express(model, 'path/to/location')

Convert a PyTorch model, e.g.:

from aifes import pytorch2aifes

pytorch2aifes.convert_to_fnn_f32_express(model, 'path/to/location')

Documentation

For a detailed documentation see here.