From 31b35c46205773bc2377bdc37e9b0bcb929968d9 Mon Sep 17 00:00:00 2001 From: Dustin Franklin Date: Wed, 10 May 2023 13:52:17 -0400 Subject: [PATCH] updated Docker scripts --- docker/build.sh | 12 +----------- docker/run.sh | 22 ---------------------- 2 files changed, 1 insertion(+), 33 deletions(-) diff --git a/docker/build.sh b/docker/build.sh index 042b6993c..7bd4c5050 100755 --- a/docker/build.sh +++ b/docker/build.sh @@ -52,19 +52,9 @@ fi echo "BASE_IMAGE=$BASE_IMAGE" echo "TAG=$TAG" - -# sanitize workspace (so extra files aren't added to the container) -rm -rf python/training/classification/data/* -rm -rf python/training/classification/models/* - -rm -rf python/training/detection/ssd/data/* -rm -rf python/training/detection/ssd/models/* - - # distro release-dependent build options source docker/containers/scripts/opencv_version.sh - - + # build the container sudo docker build -t $TAG -f Dockerfile \ --build-arg BASE_IMAGE=$BASE_IMAGE \ diff --git a/docker/run.sh b/docker/run.sh index 56aeb6916..7fdf8b773 100755 --- a/docker/run.sh +++ b/docker/run.sh @@ -56,28 +56,6 @@ RECOGNIZER_DIR="python/www/recognizer" DOCKER_ROOT="/jetson-inference" # where the project resides inside docker -# check if we need to download models -#SIZE_MODELS=$(du -sb $NETWORKS_DIR | cut -f 1) - -#echo "size of $NETWORKS_DIR: $SIZE_MODELS bytes" - -#if [[ $SIZE_MODELS -lt 204800 ]]; then # some text files come with the repo (~78KB), so check for a bit more than that -# sudo apt-get update -# sudo apt-get install dialog -# echo "Models have not yet been downloaded, running model downloader tool now..." -# cd tools -# ./download-models.sh -# cd ../ -#fi - -# check for pytorch-ssd base model -SSD_BASE_MODEL="$DETECTION_DIR/models/mobilenet-v1-ssd-mp-0_675.pth" - -if [ ! -f "$SSD_BASE_MODEL" ]; then - echo "Downloading pytorch-ssd base model..." - wget --quiet --show-progress --progress=bar:force:noscroll --no-check-certificate https://nvidia.box.com/shared/static/djf5w54rjvpqocsiztzaandq1m3avr7c.pth -O $SSD_BASE_MODEL -fi - # generate mount commands DATA_VOLUME="\ --volume $PWD/data:$DOCKER_ROOT/data \