glenn-jocher
commited on
Commit
•
8b38e6f
1
Parent(s):
1b1681b
update dataset comments
Browse files- data/coco.yaml +3 -3
- data/coco128.yaml +6 -6
- data/get_coco2017.sh +3 -2
- data/get_voc.sh +2 -1
- data/voc.yaml +5 -4
data/coco.yaml
CHANGED
@@ -1,13 +1,13 @@
|
|
1 |
# COCO 2017 dataset http://cocodataset.org
|
2 |
# Download command: bash yolov5/data/get_coco2017.sh
|
3 |
-
# Train command: python train.py --data
|
4 |
-
#
|
5 |
# /parent_folder
|
6 |
# /coco
|
7 |
# /yolov5
|
8 |
|
9 |
|
10 |
-
# train and val
|
11 |
train: ../coco/train2017.txt # 118k images
|
12 |
val: ../coco/val2017.txt # 5k images
|
13 |
test: ../coco/test-dev2017.txt # 20k images for submission to https://competitions.codalab.org/competitions/20794
|
|
|
1 |
# COCO 2017 dataset http://cocodataset.org
|
2 |
# Download command: bash yolov5/data/get_coco2017.sh
|
3 |
+
# Train command: python train.py --data coco.yaml
|
4 |
+
# Default dataset location is next to /yolov5:
|
5 |
# /parent_folder
|
6 |
# /coco
|
7 |
# /yolov5
|
8 |
|
9 |
|
10 |
+
# train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/]
|
11 |
train: ../coco/train2017.txt # 118k images
|
12 |
val: ../coco/val2017.txt # 5k images
|
13 |
test: ../coco/test-dev2017.txt # 20k images for submission to https://competitions.codalab.org/competitions/20794
|
data/coco128.yaml
CHANGED
@@ -1,15 +1,15 @@
|
|
1 |
# COCO 2017 dataset http://cocodataset.org - first 128 training images
|
2 |
-
# Download command: python -c "from yolov5.utils.google_utils import
|
3 |
-
# Train command: python train.py --data
|
4 |
-
#
|
5 |
# /parent_folder
|
6 |
# /coco128
|
7 |
# /yolov5
|
8 |
|
9 |
|
10 |
-
# train and val
|
11 |
-
train: ../coco128/images/train2017/
|
12 |
-
val: ../coco128/images/train2017/
|
13 |
|
14 |
# number of classes
|
15 |
nc: 80
|
|
|
1 |
# COCO 2017 dataset http://cocodataset.org - first 128 training images
|
2 |
+
# Download command: python -c "from yolov5.utils.google_utils import *; gdrive_download('1n_oKgR81BJtqk75b00eAjdv03qVCQn2f', 'coco128.zip')"
|
3 |
+
# Train command: python train.py --data coco128.yaml
|
4 |
+
# Default dataset location is next to /yolov5:
|
5 |
# /parent_folder
|
6 |
# /coco128
|
7 |
# /yolov5
|
8 |
|
9 |
|
10 |
+
# train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/]
|
11 |
+
train: ../coco128/images/train2017/ # 128 images
|
12 |
+
val: ../coco128/images/train2017/ # 128 images
|
13 |
|
14 |
# number of classes
|
15 |
nc: 80
|
data/get_coco2017.sh
CHANGED
@@ -1,12 +1,13 @@
|
|
1 |
#!/bin/bash
|
2 |
# COCO 2017 dataset http://cocodataset.org
|
3 |
# Download command: bash yolov5/data/get_coco2017.sh
|
4 |
-
# Train command: python train.py --data
|
5 |
-
#
|
6 |
# /parent_folder
|
7 |
# /coco
|
8 |
# /yolov5
|
9 |
|
|
|
10 |
# Download labels from Google Drive, accepting presented query
|
11 |
filename="coco2017labels.zip"
|
12 |
fileid="1cXZR_ckHki6nddOmcysCuuJFM--T-Q6L"
|
|
|
1 |
#!/bin/bash
|
2 |
# COCO 2017 dataset http://cocodataset.org
|
3 |
# Download command: bash yolov5/data/get_coco2017.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 |
|
10 |
+
|
11 |
# Download labels from Google Drive, accepting presented query
|
12 |
filename="coco2017labels.zip"
|
13 |
fileid="1cXZR_ckHki6nddOmcysCuuJFM--T-Q6L"
|
data/get_voc.sh
CHANGED
@@ -1,11 +1,12 @@
|
|
1 |
# PASCAL VOC dataset http://host.robots.ox.ac.uk/pascal/VOC/
|
2 |
# Download command: bash ./data/get_voc.sh
|
3 |
# Train command: python train.py --data voc.yaml
|
4 |
-
#
|
5 |
# /parent_folder
|
6 |
# /VOC
|
7 |
# /yolov5
|
8 |
|
|
|
9 |
start=`date +%s`
|
10 |
|
11 |
# handle optional download dir
|
|
|
1 |
# PASCAL VOC dataset http://host.robots.ox.ac.uk/pascal/VOC/
|
2 |
# Download command: bash ./data/get_voc.sh
|
3 |
# Train command: python train.py --data voc.yaml
|
4 |
+
# Default dataset location is next to /yolov5:
|
5 |
# /parent_folder
|
6 |
# /VOC
|
7 |
# /yolov5
|
8 |
|
9 |
+
|
10 |
start=`date +%s`
|
11 |
|
12 |
# handle optional download dir
|
data/voc.yaml
CHANGED
@@ -1,14 +1,15 @@
|
|
1 |
# PASCAL VOC dataset http://host.robots.ox.ac.uk/pascal/VOC/
|
2 |
# Download command: bash ./data/get_voc.sh
|
3 |
# Train command: python train.py --data voc.yaml
|
4 |
-
#
|
5 |
# /parent_folder
|
6 |
# /VOC
|
7 |
# /yolov5
|
8 |
|
9 |
-
|
10 |
-
train:
|
11 |
-
|
|
|
12 |
|
13 |
# number of classes
|
14 |
nc: 20
|
|
|
1 |
# PASCAL VOC dataset http://host.robots.ox.ac.uk/pascal/VOC/
|
2 |
# Download command: bash ./data/get_voc.sh
|
3 |
# Train command: python train.py --data voc.yaml
|
4 |
+
# Default dataset location is next to /yolov5:
|
5 |
# /parent_folder
|
6 |
# /VOC
|
7 |
# /yolov5
|
8 |
|
9 |
+
|
10 |
+
# train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/]
|
11 |
+
train: ../VOC/images/train/ # 16551 images
|
12 |
+
val: ../VOC/images/val/ # 4952 images
|
13 |
|
14 |
# number of classes
|
15 |
nc: 20
|