onnx-converter 848 B

12345678910111213141516171819202122232425262728293031
  1. #!/bin/bash
  2. set -e
  3. root=$(cd $(dirname ${0})/..; pwd)
  4. build=${root}/build
  5. tools=${root}/tools
  6. third_party=${root}/third_party
  7. identifier=onnxmltools
  8. repository=https://github.com/onnx/${identifier}.git
  9. mkdir -p ${third_party}
  10. if [ -d "${third_party}/${identifier}" ]; then
  11. git -C "${third_party}/${identifier}" fetch -p --quiet
  12. git -C "${third_party}/${identifier}" reset --quiet --hard origin/master
  13. else
  14. echo "Clone ${repository}..."
  15. git -C "${third_party}" clone --recursive ${repository}
  16. fi
  17. export PYTHONUSERBASE=${build}/third_party/pypi/${identifier}
  18. export PATH=$PATH:${PYTHONUSERBASE}/bin
  19. rm -rf ${PYTHONUSERBASE}
  20. pip install --quiet --user coremltools
  21. pip install --quiet --user tensorflow
  22. pip install --quiet --user keras
  23. pip install --quiet --user ${third_party}/${identifier}
  24. python ${tools}/onnx-converter.py $@