Spaces:
Sleeping
Sleeping
# YOLOv5 π by Ultralytics, AGPL-3.0 license | |
# Download COCO 2017 dataset http://cocodataset.org | |
# Example usage: bash data/scripts/get_coco.sh | |
# parent | |
# βββ yolov5 | |
# βββ datasets | |
# βββ coco β downloads here | |
# Arguments (optional) Usage: bash data/scripts/get_coco.sh --train --val --test --segments | |
if [ "$#" -gt 0 ]; then | |
for opt in "$@"; do | |
case "${opt}" in | |
--train) train=true ;; | |
--val) val=true ;; | |
--test) test=true ;; | |
--segments) segments=true ;; | |
esac | |
done | |
else | |
train=true | |
val=true | |
test=false | |
segments=false | |
fi | |
# Download/unzip labels | |
d='../datasets' # unzip directory | |
url=https://github.com/ultralytics/yolov5/releases/download/v1.0/ | |
if [ "$segments" == "true" ]; then | |
f='coco2017labels-segments.zip' # 168 MB | |
else | |
f='coco2017labels.zip' # 46 MB | |
fi | |
echo 'Downloading' $url$f ' ...' | |
curl -L $url$f -o $f -# && unzip -q $f -d $d && rm $f & | |
# Download/unzip images | |
d='../datasets/coco/images' # unzip directory | |
url=http://images.cocodataset.org/zips/ | |
if [ "$train" == "true" ]; then | |
f='train2017.zip' # 19G, 118k images | |
echo 'Downloading' $url$f '...' | |
curl -L $url$f -o $f -# && unzip -q $f -d $d && rm $f & | |
fi | |
if [ "$val" == "true" ]; then | |
f='val2017.zip' # 1G, 5k images | |
echo 'Downloading' $url$f '...' | |
curl -L $url$f -o $f -# && unzip -q $f -d $d && rm $f & | |
fi | |
if [ "$test" == "true" ]; then | |
f='test2017.zip' # 7G, 41k images (optional) | |
echo 'Downloading' $url$f '...' | |
curl -L $url$f -o $f -# && unzip -q $f -d $d && rm $f & | |
fi | |
wait # finish background tasks | |