mirror of
https://github.com/ultralytics/yolov5.git
synced 2025-06-03 14:49:29 +08:00
b61143c7e5
* added imagenet small versions 10,100 and 1000 * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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
|