--- a +++ b/data/scripts/get_imagenet10.sh @@ -0,0 +1,29 @@ +#!/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