22 lines
738 B
YAML
22 lines
738 B
YAML
|
# PASCAL VOC dataset http://host.robots.ox.ac.uk/pascal/VOC/
|
||
|
|
# Train command: python train.py --data voc.yaml
|
||
|
|
# Default dataset location is next to /yolov5:
|
||
|
|
# /parent_folder
|
||
|
|
# /VOC
|
||
|
|
# /yolov5
|
||
|
|
|
||
|
|
|
||
|
|
# download command/URL (optional)
|
||
|
|
download: bash data/scripts/get_voc.sh
|
||
|
|
|
||
|
|
# train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/]
|
||
|
|
train: ../VOC/images/train/ # 16551 images
|
||
|
|
val: ../VOC/images/val/ # 4952 images
|
||
|
|
|
||
|
|
# number of classes
|
||
|
|
nc: 20
|
||
|
|
|
||
|
|
# class names
|
||
|
|
names: [ 'aeroplane', 'bicycle', 'bird', 'boat', 'bottle', 'bus', 'car', 'cat', 'chair', 'cow', 'diningtable', 'dog',
|
||
|
|
'horse', 'motorbike', 'person', 'pottedplant', 'sheep', 'sofa', 'train', 'tvmonitor' ]
|