/ONNX-Python-Examples

ONNX Python Examples

Primary LanguageDockerfileMIT LicenseMIT

ONNX Python Examples

Introduction

ONNX Python examples showing how to create and modify ONNX models using ONNX Python API.

Usages

Build Docker Image

$ docker build -f docker/onnx.Dockerfile --no-cache --tag=onnx:0.0.1 .

Run Docker Container

$ docker run -it --rm --ipc=host -v $(pwd):/mnt onnx:0.0.1

Run Examples

Create Neural Network

Create a dummy convolutional neural network from scratch using ONNX Python API.

$ python create_convnet.py

Run Neural Network

Run the dummy convolutional neural network using ONNX Runtime.

$ python run_convnet.py

Modify Neural Network

Modify the dummy convolutional neural network using ONNX Python API.

$ python modify_convnet.py

References