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.

28 lines
963B

  1. #!/bin/bash
  2. # COCO 2017 dataset http://cocodataset.org
  3. # Download command: bash data/scripts/get_coco.sh
  4. # Train command: python train.py --data coco.yaml
  5. # Default dataset location is next to /yolov5:
  6. # /parent_folder
  7. # /coco
  8. # /yolov5
  9. # Download/unzip labels
  10. d='../' # unzip directory
  11. url=https://github.com/ultralytics/yolov5/releases/download/v1.0/
  12. f='coco2017labels.zip' # or 'coco2017labels-segments.zip', 68 MB
  13. echo 'Downloading' $url$f ' ...'
  14. curl -L $url$f -o $f && unzip -q $f -d $d && rm $f & # download, unzip, remove in background
  15. # Download/unzip images
  16. d='../coco/images' # unzip directory
  17. url=http://images.cocodataset.org/zips/
  18. f1='train2017.zip' # 19G, 118k images
  19. f2='val2017.zip' # 1G, 5k images
  20. f3='test2017.zip' # 7G, 41k images (optional)
  21. for f in $f1 $f2; do
  22. echo 'Downloading' $url$f '...'
  23. curl -L $url$f -o $f && unzip -q $f -d $d && rm $f & # download, unzip, remove in background
  24. done
  25. wait # finish background tasks