KB-VQA / models /yolov5 /data /scripts /get_imagenet1000.sh
m7mdal7aj's picture
uploading a fresh copy of the detection models
92894b3 verified
raw
history blame
742 Bytes
#!/bin/bash
# YOLOv5 πŸš€ by Ultralytics, AGPL-3.0 license
# Download ILSVRC2012 ImageNet dataset https://image-net.org
# Example usage: bash data/scripts/get_imagenet.sh
# parent
# β”œβ”€β”€ yolov5
# └── datasets
# └── imagenet ← downloads here
# Arguments (optional) Usage: bash data/scripts/get_imagenet.sh --train --val
if [ "$#" -gt 0 ]; then
for opt in "$@"; do
case "${opt}" in
--train) train=true ;;
--val) val=true ;;
esac
done
else
train=true
val=true
fi
# Make dir
d='../datasets/imagenet1000' # unzip directory
mkdir -p $d && cd $d
# Download/unzip train
wget https://github.com/ultralytics/yolov5/releases/download/v1.0/imagenet1000.zip
unzip imagenet1000.zip && rm imagenet1000.zip