added non-interactive build option

CMakeLists.txt

@@ -25,6 +25,7 @@ endif()
 # setup tensorRT flags
 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -Wno-write-strings -Wno-deprecated-declarations")	# -std=c++14 
 set(BUILD_DEPS "YES" CACHE BOOL "If YES, will install dependencies into sandbox.  Automatically reset to NO after dependencies are installed.")
+set(BUILD_INTERACTIVE "YES" CACHE BOOL "If NO, will download/install the default DNN models without prompting the user, and skip installation of PyTorch.")
 
 
 # copy configuration tools to build dir
@@ -37,8 +38,9 @@ file(COPY "tools/install-pytorch.rc" DESTINATION ${PROJECT_BINARY_DIR})
 # if this is the first time running cmake, perform pre-build dependency install script (or if the user manually triggers re-building the dependencies)
 if( ${BUILD_DEPS} )
 	message("-- Launching pre-build dependency installer script...")
+	message("-- Build interactive:  ${BUILD_INTERACTIVE}")
 
-	execute_process(COMMAND sh ../CMakePreBuild.sh
+	execute_process(COMMAND sh ../CMakePreBuild.sh ${BUILD_INTERACTIVE}
 				WORKING_DIRECTORY ${PROJECT_BINARY_DIR}
 				RESULT_VARIABLE PREBUILD_SCRIPT_RESULT)
 

CMakePreBuild.sh

@@ -2,9 +2,11 @@
 # this script is automatically run from CMakeLists.txt
 
 BUILD_ROOT=$PWD
+BUILD_INTERACTIVE=$1
 
 echo "[Pre-build]  dependency installer script running..."
-echo "[Pre-build]  build root directory:   $BUILD_ROOT"
+echo "[Pre-build]  build root directory: $BUILD_ROOT"
+echo "[Pre-build]  build interactive:    $BUILD_INTERACTIVE"
 echo " "
 
 
@@ -22,11 +24,11 @@ sudo apt-get update
 
 
 # run the model downloader
-./download-models.sh
+./download-models.sh $BUILD_INTERACTIVE
 
 
 # run the pytorch installer
-./install-pytorch.sh
+./install-pytorch.sh $BUILD_INTERACTIVE
 
 
 echo "[Pre-build]  Finished CMakePreBuild script"

tools/download-models.sh

@@ -25,6 +25,7 @@ APP_TITLE="Hello AI World (jetson-inference)"
 OUTPUT_DIR="../data/networks"
 LOG="[jetson-inference] "
 WGET_QUIET="--quiet"
+BUILD_INTERACTIVE=$1
 
 
 #
@@ -588,6 +589,34 @@ function install_deb_package()
 }
 
 
+#
+# non-interactive mode
+#
+echo "$LOG BUILD_INTERACTVE=$BUILD_INTERACTIVE"
+
+if [[ "$BUILD_INTERACTIVE" != "YES" ]]; then
+
+	echo "$LOG Downloading default models..."
+
+	download_googlenet
+	download_resnet18
+
+	download_ssd_mobilenet_v2
+	download_pednet
+	download_facenet
+	download_detectnet_coco_dog
+
+	download_fcn_resnet18_cityscapes_512x256
+	download_fcn_resnet18_cityscapes_1024x512
+	download_fcn_resnet18_deepscene_576x320
+	download_fcn_resnet18_mhp_512x320
+	download_fcn_resnet18_pascal_voc_320x320
+	download_fcn_resnet18_sun_rgbd_512x400
+
+	exit_message 0
+fi
+
+
 # check for dialog package
 install_deb_package "dialog" FOUND_DIALOG
 echo "$LOG FOUND_DIALOG=$FOUND_DIALOG"

tools/install-pytorch.sh

@@ -24,6 +24,7 @@
 APP_TITLE="Hello AI World (jetson-inference)"
 LOG="[jetson-inference] "
 WGET_QUIET="--quiet"
+BUILD_INTERACTIVE=$1
 
 
 #
@@ -353,6 +354,17 @@ function check_L4T_version()
 }
 
 
+#
+# non-interactive mode
+#
+echo "$LOG BUILD_INTERACTVE=$BUILD_INTERACTIVE"
+
+if [[ "$BUILD_INTERACTIVE" != "YES" ]]; then
+	echo "$LOG non-interactive mode, skipping PyTorch install..."
+	exit_message 0
+fi
+
+
 # check for dialog package
 install_deb_package "dialog" FOUND_DIALOG
 echo "$LOG FOUND_DIALOG=$FOUND_DIALOG"
@@ -371,9 +383,9 @@ while true; do
 
 	packages_selected=$(dialog --backtitle "$APP_TITLE" \
 							  --title "PyTorch Installer (L4T R$JETSON_L4T_VERSION)" \
-							  --cancel-label "Quit" \
+							  --cancel-label "Skip" \
 							  --colors \
-							  --checklist "If you want to train DNN models on your Jetson, this tool will download and install PyTorch.  Select the desired versions of pre-built packages below, or see \Zbhttp://eLinux.org/Jetson_Zoo\Zn for instructions to build from source. \n\nYou can skip this step and select Quit if you don't want to install PyTorch.\n\n\ZbKeys:\Zn\n  ↑↓ Navigate Menu\n  Space to Select \n  Enter to Continue\n\n\ZbPackages to Install:\Zn" 20 80 2 \
+							  --checklist "If you want to train DNN models on your Jetson, this tool will download and install PyTorch.  Select the desired versions of pre-built packages below, or see \Zbhttp://eLinux.org/Jetson_Zoo\Zn for instructions to build from source. \n\nYou can skip this step and select Skip if you don't want to install PyTorch.\n\n\ZbKeys:\Zn\n  ↑↓ Navigate Menu\n  Space to Select \n  Enter to Continue\n\n\ZbPackages to Install:\Zn" 20 80 2 \
 							  --output-fd 1 \
 							  1 "PyTorch v1.1.0 for Python 2.7" off \
 							  2 "PyTorch v1.1.0 for Python 3.6" off \