|
|
|
@ -1,15 +1,15 @@
|
|
|
|
|
# COCO 2017 dataset http://cocodataset.org - first 128 training images
|
|
|
|
|
# Download command: python -c "from yolov5.utils.google_utils import gdrive_download; gdrive_download('1n_oKgR81BJtqk75b00eAjdv03qVCQn2f','coco128.zip')"
|
|
|
|
|
# Train command: python train.py --data ./data/coco128.yaml
|
|
|
|
|
# Dataset should be placed next to yolov5 folder:
|
|
|
|
|
# Download command: python -c "from yolov5.utils.google_utils import *; gdrive_download('1n_oKgR81BJtqk75b00eAjdv03qVCQn2f', 'coco128.zip')"
|
|
|
|
|
# Train command: python train.py --data coco128.yaml
|
|
|
|
|
# Default dataset location is next to /yolov5:
|
|
|
|
|
# /parent_folder
|
|
|
|
|
# /coco128
|
|
|
|
|
# /yolov5
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# train and val datasets (image directory or *.txt file with image paths)
|
|
|
|
|
train: ../coco128/images/train2017/
|
|
|
|
|
val: ../coco128/images/train2017/
|
|
|
|
|
# train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/]
|
|
|
|
|
train: ../coco128/images/train2017/ # 128 images
|
|
|
|
|
val: ../coco128/images/train2017/ # 128 images
|
|
|
|
|
|
|
|
|
|
# number of classes
|
|
|
|
|
nc: 80
|
|
|
|
|