/mxzoo2onnx

Primary LanguageJupyter Notebook

mxzoo2onnx

A sample Jupyter notebook to explain how to export MXNet pretrained model to ONNX format.

Target platform

  • linux/amd64 (GPU not needed)

Prerequisites

  • Docker
    • Configure Docker can be run by non-root user. Please refer this article.

Installation

  1. Clone this repository.

    git clone https://github.com/MACNICA-CLAVIS-NV/mxzoo2onnx

  2. Build the Docker cotainer image contains the required softwares for this Jupyter notebook.

    ./docker_build.sh

Usage

  1. Run a container from the build image.

    ./docker_run.sh

  2. Launch Jupyter Lab with the instruction displyed at the above step.

  3. Open the mxnet_model_zoo_to_onnx.ipynb notebook from the /mxzoo2onnx path in the Jupyter file browser.