You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.7 KiB
60 lines
1.7 KiB
#!/bin/bash |
|
# Ultralytics YOLO 🚀, AGPL-3.0 license |
|
# Download COCO 2017 dataset http://cocodataset.org |
|
# Example usage: bash data/scripts/get_coco.sh |
|
# parent |
|
# ├── ultralytics |
|
# └── 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 ;; |
|
--sama) sama=true ;; |
|
esac |
|
done |
|
else |
|
train=true |
|
val=true |
|
test=false |
|
segments=false |
|
sama=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' # 169 MB |
|
elif [ "$sama" == "true" ]; then |
|
f='coco2017labels-segments-sama.zip' # 199 MB https://www.sama.com/sama-coco-dataset/ |
|
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
|
|
|