ONNX TensorFlow

Name:flytekitplugins-onnxtensorflow
Version:0.0.0+develop
Author:admin@flyte.org
Provides: flytekitplugins.onnxtensorflow
Requires: flytekit>=1.3.0b2,<2.0.0
tf2onnx>=1.9.3
tensorflow>=2.7.0
Python:>=3.9
License:apache2
Source Code: https://github.com/flyteorg/flytekit/tree/master/plugins/flytekit-onnx-tensorflow
  • Intended Audience :: Science/Research
  • Intended Audience :: Developers
  • License :: OSI Approved :: Apache Software License
  • Programming Language :: Python :: 3.9
  • Programming Language :: Python :: 3.10
  • Topic :: Scientific/Engineering
  • Topic :: Scientific/Engineering :: Artificial Intelligence
  • Topic :: Software Development
  • Topic :: Software Development :: Libraries
  • Topic :: Software Development :: Libraries :: Python Modules

This plugin allows you to generate ONNX models from your TensorFlow Keras models.

To install the plugin, run the following command:

pip install flytekitplugins-onnxtensorflow