netron/tools/onnx-generate

#!/bin/bash

mkdir -p ../third_party

[email protected]:onnx/onnx.git

if [ -d "../third_party/onnx" ]; then
    pushd "../third_party/onnx" > /dev/null
    echo "Fetch ${repository}..."
    git fetch -p
    echo "Reset ${repository}..."
    git reset --hard origin/master
    popd > /dev/null
else
    echo "Clone ${repository}..."
    pushd "../third_party" > /dev/null
    git clone --recursive ${repository}
    popd > /dev/null
fi

export ONNX_ML=1

pushd "../third_party/onnx" > /dev/null
sudo -E python setup.py build install
popd > /dev/null

echo "Generate '../src/onnx.js'"
../node_modules/protobufjs/bin/pbjs -t static-module -w closure -r onnx -o ../src/onnx.js ../third_party/onnx/onnx/onnx-ml.proto

echo "Generate '../src/onnx-operator.json'"
python onnx-operator-json.py

echo "Generate '../src/onnx.py'"
protoc --python_out=../src --proto_path ../third_party/onnx/onnx onnx-ml.proto