model downloader tool

Media-4-Machine-Laboratory · Jun 11, 2019 · eb15cc0 · eb15cc0
1 parent 6532311
commit eb15cc0
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 106 deletions.
diff --git a/CMakePreBuild.sh b/CMakePreBuild.sh
@@ -21,104 +21,8 @@ sudo apt-get install -y libopencv-calib3d-dev libopencv-dev
 sudo apt-get update
 
 
-# libgstreamer-plugins-base1.0-dev
+# run the model downloader
+./download-models.sh
 
-#sudo rm /usr/lib/aarch64-linux-gnu/libGL.so
-#sudo ln -s /usr/lib/aarch64-linux-gnu/tegra/libGL.so /usr/lib/aarch64-linux-gnu/libGL.so
-
-
-# uncomment to download Alexnet (220MB)
-#wget http://dl.caffe.berkeleyvision.org/bvlc_alexnet.caffemodel
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/5j264j7mky11q8emy4q14w3r8hl5v6zh.caffemodel' -O bvlc_alexnet.caffemodel
-mv bvlc_alexnet.caffemodel ../data/networks
-
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/c84wp3axbtv4e2gybn40jprdquav9azm.prototxt' -O alexnet.prototxt
-mv alexnet.prototxt ../data/networks
-
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/o0w0sl3obqxj21u09c0cwzw4khymz7hh.prototxt' -O alexnet_noprob.prototxt
-mv alexnet_noprob.prototxt ../data/networks
-
-# GoogleNet (bvlc site was behaving slowly, so enabled mirror on nvidia.box.com instead)
-#wget http://dl.caffe.berkeleyvision.org/bvlc_googlenet.caffemodel
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/at8b1105ww1c5h7p30j5ko8qfnxrs0eg.caffemodel' -O bvlc_googlenet.caffemodel
-mv bvlc_googlenet.caffemodel ../data/networks
-
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/5z3l76p8ap4n0o6rk7lyasdog9f14gc7.prototxt' -O googlenet.prototxt
-mv googlenet.prototxt ../data/networks
-
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/ue8qrqtglu36andbvobvaaj8egxjaoli.prototxt' -O googlenet_noprob.prototxt
-mv googlenet_noprob.prototxt ../data/networks
-
-# GoogleNet, ILSVR12 subset
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/zb8i3zcg39sdjjxfty7o5935hpbd64y4.gz' -O GoogleNet-ILSVRC12-subset.tar
-tar -xzvf GoogleNet-ILSVRC12-subset.tar -C ../data/networks
-
-
-# DetectNet's  (uncomment to download)
-#wget --no-check-certificate 'https://nvidia.box.com/shared/static/xe6wo1o8qiqykfx8umuu0ki9idp0f92p.prototxt' -O detectnet.prototxt
-#mv detectnet.prototxt ../data/networks
-
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/0wbxo6lmxfamm1dk90l8uewmmbpbcffb.gz' -O ped-100.tar.gz
-tar -xzvf ped-100.tar.gz -C ../data/networks
-
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/r3bq08qh7zb0ap2lf4ysjujdx64j8ofw.gz' -O multiped-500.tar.gz
-tar -xzvf multiped-500.tar.gz -C ../data/networks
-
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/wjitc00ef8j6shjilffibm6r2xxcpigz.gz' -O facenet-120.tar.gz
-tar -xzvf facenet-120.tar.gz -C ../data/networks
-
-
-# COCO (DetectNet)
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/xi71hlsht5b0y66loeg73rxfa73q561s.gz' -O DetectNet-COCO-Airplane.tar.gz
-tar -xzvf DetectNet-COCO-Airplane.tar.gz -C ../data/networks
-
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/8bhm91o9yldpf97dcz5d0welgmjy7ucw.gz' -O DetectNet-COCO-Bottle.tar.gz
-tar -xzvf DetectNet-COCO-Bottle.tar.gz -C ../data/networks
-
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/fq0m0en5mmssiizhs9nxw3xtwgnoltf2.gz' -O DetectNet-COCO-Chair.tar.gz
-tar -xzvf DetectNet-COCO-Chair.tar.gz -C ../data/networks
-
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/3qdg3z5qvl8iwjlds6bw7bwi2laloytu.gz' -O DetectNet-COCO-Dog.tar.gz
-tar -xzvf DetectNet-COCO-Dog.tar.gz -C ../data/networks
-
-
-# SSD (UFF)
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/zp9cmlq4i5ho3sh6gabt7dkhzuxd8s55.gz' -O SSD.tar.gz
-tar -xzvf SSD.tar.gz -C ../data/networks
-
-
-# Segmentation Nets (uncomment to download)
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/xj20b6qopfwkkpqm12ffiuaekk6bs8op.gz' -O FCN-Alexnet-Pascal-VOC.tar.gz
-tar -xzvf FCN-Alexnet-Pascal-VOC.tar.gz -C ../data/networks
-
-#wget --no-check-certificate 'https://nvidia.box.com/shared/static/u5ey2ws0nbtzyqyftkuqazx1honw6wry.gz' -O FCN-Alexnet-SYNTHIA-CVPR16.tar.gz
-#tar -xzvf FCN-Alexnet-SYNTHIA-CVPR16.tar.gz -C ../data/networks
-
-#wget --no-check-certificate 'https://nvidia.box.com/shared/static/ydgmqgdhbvul6q9avoc9flxr3fdoa8pw.gz' -O FCN-Alexnet-SYNTHIA-Summer-HD.tar.gz
-#tar -xzvf FCN-Alexnet-SYNTHIA-Summer-HD.tar.gz -C ../data/networks
-
-#wget --no-check-certificate 'https://nvidia.box.com/shared/static/vbk5ofu1x2hwp9luanbg4o0vrfub3a7j.gz' -O FCN-Alexnet-SYNTHIA-Summer-SD.tar.gz
-#tar -xzvf FCN-Alexnet-SYNTHIA-Summer-SD.tar.gz -C ../data/networks
-
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/mh121fvmveemujut7d8c9cbmglq18vz3.gz' -O FCN-Alexnet-Cityscapes-HD.tar.gz
-tar -xzvf FCN-Alexnet-Cityscapes-HD.tar.gz -C ../data/networks
-
-#wget --no-check-certificate 'https://nvidia.box.com/shared/static/pa5d338t9ntca5chfbymnur53aykhall.gz' -O FCN-Alexnet-Cityscapes-SD.tar.gz
-#tar -xzvf FCN-Alexnet-Cityscapes-SD.tar.gz -C ../data/networks
-
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/y1mzlwkmytzwg2m7akt7tcbsd33f9opz.gz' -O FCN-Alexnet-Aerial-FPV-720p.tar.gz
-tar -xzvf FCN-Alexnet-Aerial-FPV-720p.tar.gz -C ../data/networks
-
-#wget --no-check-certificate 'https://nvidia.box.com/shared/static/4z5lmlja13blj3mdn6vesrft57p30446.gz' -O FCN-Alexnet-Aerial-FPV-4ch-720p.tar.gz
-#tar -xzvf FCN-Alexnet-Aerial-FPV-4ch-720p.tar.gz -C ../data/networks
-
-
-# Deep Homography
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/nlqbsdnt76y0nmkwdzxkg4zbvhk4bidh.gz' -O Deep-Homography-COCO.tar.gz
-tar -xzvf Deep-Homography-COCO.tar.gz -C ../data/networks
-
-# Super Resolution
-wget --no-check-certificate 'https://nvidia.box.com/shared/static/a99l8ttk21p3tubjbyhfn4gh37o45rn8.gz' -O Super-Resolution-BSD500.tar.gz
-tar -xzvf Super-Resolution-BSD500.tar.gz -C ../data/networks
 
 echo "[Pre-build]  Finished CMakePreBuild script"
diff --git a/data/images/fontmapA.png b/data/images/fontmapA.png
diff --git a/data/images/fontmapB.png b/data/images/fontmapB.png
diff --git a/tools/download-models.sh b/tools/download-models.sh
@@ -34,8 +34,9 @@ function exit_message()
 {
 	echo " "
 	echo "$LOG to run this tool again, use the following commands:"
-	echo "             $ cd <jetson-inference>/tools"
-	echo "             $ ./download-models.sh"
+	echo " "
+	echo "    $ cd <jetson-inference>/tools"
+	echo "    $ ./download-models.sh"
 	echo " "
 
 	exit $1
@@ -140,7 +141,7 @@ function attempt_download_archive()
 		return $wget_status
 	fi
 
-	tar -xzvf $filename -C $OUTPUT_DIR
+	tar -xzf $filename -C $OUTPUT_DIR
 
 	local tar_status=$?
 
@@ -479,14 +480,14 @@ while true; do
 							  13 "   > DetectNet-COCO-Bottle      (29 MB)" off \
 							  14 "   > DetectNet-COCO-Chair       (29 MB)" off \
 							  15 "   > DetectNet-COCO-Airplane    (29 MB)" off \
-							  16 "\ZbSemantic Segmentation - all     (477 MB)\Zn" off \
+							  16 "\ZbSemantic Segmentation - all     (1.4 GB)\Zn" off \
 							  17 "   > FCN-Alexnet-Cityscapes-SD  (235 MB)" off \
 							  18 "   > FCN-Alexnet-Cityscapes-HD  (235 MB)" off \
 							  19 "   > FCN-Alexnet-Aerial-FPV     (7 MB)" on \
-							  20 "   > FCN-Alexnet-Pascal-VOC     (?? MB)" off \
-							  21 "   > FCN-Alexnet-Synthia-CVPR   (?? MB)" off \
-							  22 "   > FCN-Alexnet-Synthia-Summer-SD (?? MB)" off \
-							  23 "   > FCN-Alexnet-Synthia-Summer-HD (?? MB)" off \
+							  20 "   > FCN-Alexnet-Pascal-VOC     (235 MB)" off \
+							  21 "   > FCN-Alexnet-Synthia-CVPR   (235 MB)" off \
+							  22 "   > FCN-Alexnet-Synthia-Summer-SD (235 MB)" off \
+							  23 "   > FCN-Alexnet-Synthia-Summer-HD (235 MB)" off \
 							  24 "\ZbImage Processing - all models   (138 MB)\Zn" off \
 							  25 "   > Deep-Homography-COCO       (137 MB)" off \
 							  26 "   > Super-Resolution-BSD500    (1 MB)" off )