Skip to content

Commit

Permalink
updated Docker scripts
Browse files Browse the repository at this point in the history
  • Loading branch information
dusty-nv committed May 10, 2023
1 parent df9a1ea commit 31b35c4
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 33 deletions.
12 changes: 1 addition & 11 deletions docker/build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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 \
Expand Down
22 changes: 0 additions & 22 deletions docker/run.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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 \
Expand Down

0 comments on commit 31b35c4

Please sign in to comment.