Update onnx-converter script

lutzroeder · Jun 17, 2018 · 052191c · 052191c
1 parent e2f66d8
commit 052191c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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 $@