diff --git a/install.sh b/install.sh
index c569148..9484407 100755
--- a/install.sh
+++ b/install.sh
@@ -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
@@ -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 $@