Add get_coco128.sh for downloading the coco128 dataset (#3047)
Browse files* Add get_coco128.sh
* Update get_coco128.sh
Co-authored-by: Glenn Jocher <[email protected]>
- data/scripts/get_coco128.sh +17 -0
data/scripts/get_coco128.sh
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/bin/bash
|
| 2 |
+
# COCO128 dataset https://www.kaggle.com/ultralytics/coco128
|
| 3 |
+
# Download command: bash data/scripts/get_coco128.sh
|
| 4 |
+
# Train command: python train.py --data coco128.yaml
|
| 5 |
+
# Default dataset location is next to /yolov5:
|
| 6 |
+
# /parent_folder
|
| 7 |
+
# /coco128
|
| 8 |
+
# /yolov5
|
| 9 |
+
|
| 10 |
+
# Download/unzip images and labels
|
| 11 |
+
d='../' # unzip directory
|
| 12 |
+
url=https://github.com/ultralytics/yolov5/releases/download/v1.0/
|
| 13 |
+
f='coco128.zip' # or 'coco2017labels-segments.zip', 68 MB
|
| 14 |
+
echo 'Downloading' $url$f ' ...'
|
| 15 |
+
curl -L $url$f -o $f && unzip -q $f -d $d && rm $f & # download, unzip, remove in background
|
| 16 |
+
|
| 17 |
+
wait # finish background tasks
|