Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 13 additions & 8 deletions install.sh
Original file line number Diff line number Diff line change
Expand Up @@ -4,11 +4,16 @@ INSTALL_PROTOC=$PWD/scripts/install_protoc.sh
MODELS_DIR=$PWD/third_party/models

PYTHON=python

if [ $# -eq 1 ]; then
PYTHON=$1
fi

INSTALL="install"
for arg do
shift
case $arg in
develop) INSTALL="develop";;
install) INSTALL="install";;
*python*) PYTHON=$arg;;
*) set -- "$@" "$arg";;
esac
done
echo $PYTHON

# install protoc
Expand All @@ -24,15 +29,15 @@ echo $PWD
echo "Installing object detection library"
echo $PROTOC
$PROTOC object_detection/protos/*.proto --python_out=.
$PYTHON setup.py install --user
echo $PYTHON setup.py $INSTALL $@
popd

pushd $MODELS_DIR/research/slim
echo $PWD
echo "Installing slim library"
$PYTHON setup.py install --user
echo $PYTHON setup.py $INSTALL $@
popd

echo "Installing tf_trt_models"
echo $PWD
$PYTHON setup.py install --user
echo $PYTHON setup.py $INSTALL $@