onnion 0.8.1

onnx compiler

onnion

Installation

From PyPI:

$ pip3 install onnion

From Dockerhub:

docker pull idein/onnion:20221014

Usage

$ onnion ssd-10-post.onnx -o ssd_post_model.py
$ python
>>> from ssd_post_model import init_graph
>>> graph = init_graph()
>>> inputs = ... # List[np.array]
>>> outputs = graph.run(*inputs)

With docker:

$ docker run --rm -it -u $UID:$GID -v $(pwd):/work idein/onnion:20221014 ssd-10-post.onnx -o ssd_post_model.py

The order of the inputs and the outputs in the run method corresponds to the order of the inputs and the outputs in the onnx graph.

See also tutorial.

Development Guide

$ poetry install