You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
742 B
Bash
30 lines
742 B
Bash
#!/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
|