2020-07-05 07:11:14 +08:00
|
|
|
# PASCAL VOC dataset http://host.robots.ox.ac.uk/pascal/VOC/
|
2020-07-05 07:50:11 +08:00
|
|
|
# Download command: bash ./data/get_voc.sh
|
2020-07-05 07:11:14 +08:00
|
|
|
# Train command: python train.py --data voc.yaml
|
|
|
|
|
# Dataset should be placed next to yolov5 folder:
|
|
|
|
|
# /parent_folder
|
|
|
|
|
# /VOC
|
|
|
|
|
# /yolov5
|
2020-07-03 05:06:59 +08:00
|
|
|
|
|
|
|
|
# train and val datasets (image directory or *.txt file with image paths)
|
|
|
|
|
train: ../VOC/images/train/
|
2020-07-05 07:11:14 +08:00
|
|
|
val: ../VOC/images/val/
|
2020-07-03 05:06:59 +08:00
|
|
|
|
|
|
|
|
# number of classes
|
|
|
|
|
nc: 20
|
|
|
|
|
|
|
|
|
|
# class names
|
2020-07-05 07:50:11 +08:00
|
|
|
names: ['aeroplane', 'bicycle', 'bird', 'boat', 'bottle', 'bus', 'car', 'cat', 'chair', 'cow', 'diningtable', 'dog',
|
|
|
|
|
'horse', 'motorbike', 'person', 'pottedplant', 'sheep', 'sofa', 'train', 'tvmonitor']
|
