ONNX-MXNet API

Overview

ONNX is an open format to represent deep learning models. With ONNX as an intermediate representation, it is easier to move models between state-of-the-art tools and frameworks for training and inference.

The mxnet.contrib.onnx package refers to the APIs and interfaces that implement ONNX model format support for Apache MXNet.

With ONNX format support for MXNet, developers can build and train models with a variety of deep learning frameworks, and import these models into MXNet to run them for inference and training using MXNet’s highly optimized engine.

.. warning:: This package contains experimental APIs and may change in the near future.

Installation Instructions

  • To use this module developers need to install ONNX, which requires the protobuf compiler to be installed separately. Please follow the instructions to install ONNX and its dependencies. MXNet currently supports ONNX v1.2.1. Once installed, you can go through the tutorials on how to use this module.

This document describes all the ONNX-MXNet APIs.

.. autosummary::
    :nosignatures:

    mxnet.contrib.onnx.onnx2mx.import_model
    mxnet.contrib.onnx.onnx2mx.import_to_gluon
    mxnet.contrib.onnx.mx2onnx.export_model

ONNX Tutorials

.. toctree::
   :maxdepth: 1

   /tutorials/onnx/super_resolution.md
   /tutorials/onnx/export_mxnet_to_onnx.md
   /tutorials/onnx/inference_on_onnx_model.md
   /tutorials/onnx/fine_tuning_gluon.md

ONNX Examples

API Reference


.. automodule:: mxnet.contrib.onnx.onnx2mx.import_model
    :members: import_model, get_model_metadata
.. automodule:: mxnet.contrib.onnx.onnx2mx.import_to_gluon
    :members: import_to_gluon
.. automodule:: mxnet.contrib.onnx.mx2onnx.export_model
    :members: export_model