From 052191c2125cb323cf03a0a837cd2e7ccb70e95c Mon Sep 17 00:00:00 2001 From: Lutz Roeder Date: Sun, 17 Jun 2018 13:12:29 -0700 Subject: [PATCH] Update onnx-converter script --- tools/onnx-converter | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tools/onnx-converter b/tools/onnx-converter index 8e638dc3e2..6d4d89d29b 100755 --- a/tools/onnx-converter +++ b/tools/onnx-converter @@ -23,6 +23,9 @@ fi export PYTHONUSERBASE=${build}/third_party/pypi/${identifier} export PATH=$PATH:${PYTHONUSERBASE}/bin rm -rf ${PYTHONUSERBASE} +pip install --quiet --user coremltools +pip install --quiet --user tensorflow +pip install --quiet --user keras pip install --quiet --user ${third_party}/${identifier} python ${tools}/onnx-converter.py $@