onnx-generate 622 B

1234567891011121314151617181920212223
  1. #!/bin/bash
  2. mkdir -p ../third_party
  3. pushd "../third_party" > /dev/null
  4. sudo rm -rf onnx
  5. git clone --recursive [email protected]:onnx/onnx.git
  6. popd > /dev/null
  7. export ONNX_ML=1
  8. pushd "../third_party/onnx" > /dev/null
  9. sudo -E python setup.py build install
  10. popd > /dev/null
  11. echo "Generate '../src/onnx.js'"
  12. ../node_modules/protobufjs/bin/pbjs -t static-module -w closure -r onnx -o ../src/onnx.js ../third_party/onnx/onnx/onnx-ml.proto
  13. echo "Generate '../src/onnx-operator.json'"
  14. python onnx-operator-json.py
  15. echo "Generate '../src/onnx.py'"
  16. protoc --python_out=../src --proto_path ../third_party/onnx/onnx onnx-ml.proto