30 lines
734 B
Bash
30 lines
734 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/imagenet10' # unzip directory
|
||
|
|
mkdir -p $d && cd $d
|
||
|
|
|
||
|
|
# Download/unzip train
|
||
|
|
wget https://github.com/ultralytics/yolov5/releases/download/v1.0/imagenet10.zip
|
||
|
|
unzip imagenet10.zip && rm imagenet10.zip
|