e9ab205ef6
* Refactor code for speed and clarity * Auto-format by https://ultralytics.com/actions * Update README.md Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> --------- Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
31 lines
766 B
Bash
31 lines
766 B
Bash
#!/bin/bash
|
|
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/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
|