Download-only setup for colab.

Author: davidbau

demo_test.sh

@@ -23,10 +23,15 @@ if [ ! -e $TEST_IMG ]; then
   wget -P $RESULT_PATH http://sceneparsing.csail.mit.edu/data/ADEChallengeData2016/images/validation/$TEST_IMG
 fi
 
+if [ -z "$DOWNLOAD_ONLY" ]
+then
+
 # Inference
 python3 -u test.py \
   --imgs $TEST_IMG \
   --cfg config/ade20k-resnet50dilated-ppm_deepsup.yaml \
   DIR $MODEL_PATH \
   TEST.result ./ \
   TEST.checkpoint epoch_20.pth
+
+fi

notebooks/DemoSegmenter.ipynb

@@ -8,8 +8,8 @@
    "source": [
     "%%bash\n",
     "!(stat -t /usr/local/lib/*/dist-packages/google/colab > /dev/null 2>&1) && exit \n",
-    "git clone https://github.com/davidbau/semantic-segmentation-pytorch.git seg_code 2>> install.log\n",
-    "/content/seg_code/demo_test.sh"
+    "git clone -b tutorial https://github.com/davidbau/semantic-segmentation-pytorch.git seg_code 2>> install.log\n",
+    "DOWNLOAD_ONLY=1 /content/seg_code/demo_test.sh"
    ]
   },
   {
@@ -144,6 +144,7 @@
   }
  ],
  "metadata": {
+  "accelerator": "GPU",
   "kernelspec": {
    "display_name": "Python 3",
    "language": "python",