Skip to content

Latest commit

 

History

History
56 lines (33 loc) · 1.11 KB

README.md

File metadata and controls

56 lines (33 loc) · 1.11 KB

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