#!/bin/bash
# Downloads the ImageNet validation data set and labels.

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
OUTPUT_DIR="${SCRIPT_DIR}/imagenet"
mkdir -p "${OUTPUT_DIR}"
cd "${OUTPUT_DIR}"

# Helper function to download and unzip a .zip file
function download_and_untar() {
  local BASE_URL=${1}
  local FILENAME=${2}

  if [[ ! -f "${FILENAME}" ]]; then
    echo "Downloading ${FILENAME} to $(pwd)"
    wget -nd -c "${BASE_URL}/${FILENAME}"
  else
    echo "Skipping download of ${FILENAME}"
  fi
  echo "Unzipping ${FILENAME}"
  tar xvf "${FILENAME}"
}

# Download the validation data set.
IMAGENET_DATA_URL="http://www.image-net.org/challenges/LSVRC/2012/nnoupb"
IMAGENET_DATA_FILE="ILSVRC2012_img_val.tar"
download_and_untar ${IMAGENET_DATA_URL} ${IMAGENET_DATA_FILE}

# Download the ground truth label for classification.
IMAGENET_LABEL_URL="http://dl.caffe.berkeleyvision.org"
IMAGENET_LABEL_FILE="caffe_ilsvrc12.tar.gz"
download_and_untar ${IMAGENET_LABEL_URL} ${IMAGENET_LABEL_FILE}