blob: 0de3e72727c90e764093edaea60d319c2f20b667 [file] [log] [blame]
#!/bin/bash
# Downloads the COCO validation data set to validate the accuracy of coco object
# detection models.
#
# Install dependencies:
# sudo apt-get install -y libfreetype6-dev libpng-dev libqhull-dev libagg-dev python3-dev pkg-config
# pip install matplotlib
# pip install cython
# pip install git+https://github.com/cocodataset/cocoapi#subdirectory=PythonAPI
#
# Setup proper matplotlib backed:
# export MPLBACKEND=agg
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
OUTPUT_DIR="${SCRIPT_DIR}/coco"
mkdir -p "${OUTPUT_DIR}"
cd "${OUTPUT_DIR}"
# Helper function to download and unzip a .zip file
function download_and_unzip() {
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}"
unzip -nq "${FILENAME}"
}
# Download the validation data set.
COCO_DATA_URL="http://images.cocodataset.org/zips"
VAL_DATA_FILE="val2017.zip"
download_and_unzip ${COCO_DATA_URL} ${VAL_DATA_FILE}
# Download the annotations.
COCO_ANN_URL="http://images.cocodataset.org/annotations"
ANN_FILE="annotations_trainval2017.zip"
download_and_unzip ${COCO_ANN_URL} ${ANN_FILE}