-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathtaco.yaml
33 lines (29 loc) · 1.53 KB
/
taco.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# Train command: python train.py --data coco.yaml
# Default dataset location is next to /yolov5:
# download command/URL (optional)
#download: bash data/scripts/get_coco.sh
# train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/]
train: ../taco/train/images
val: ../taco/val/images
test: ../taco/test/images
# number of classes
nc: 59 # nc: 8
# class names
names: ['Glass bottle', 'Meal carton', 'Other carton',
'Clear plastic bottle', 'Plastic bottle cap', 'Drink can',
'Food Can', 'Other plastic bottle', 'Pop tab', 'Aerosol',
'Glass cup', 'Other plastic wrapper', 'Styrofoam piece',
'Plastic film', 'Other plastic', 'Drink carton',
'Metal bottle cap', 'Disposable food container', 'Normal paper',
'Paper cup', 'Cigarette', 'Single-use carrier bag', 'Tissues',
'Toilet tube', 'Crisp packet', 'Plastic lid', 'Metal lid',
'Egg carton', 'Plastic straw', 'Paper bag',
'Disposable plastic cup', 'Broken glass', 'Plastic utensils',
'Glass jar', 'Food waste', 'Squeezable tube', 'Spread tub', 'Shoe',
'Garbage bag', 'Aluminium foil', 'Six pack rings', 'Foam cup',
'Paper straw', 'Corrugated carton', 'Unlabeled litter',
'Aluminium blister pack', 'Battery', 'Rope & strings',
'Other plastic container', 'Polypropylene bag', 'Scrap metal',
'Magazine paper', 'Pizza box', 'Plastic glooves', 'Wrapping paper',
'Carded blister pack', 'Foam food container', 'Tupperware',
'Other plastic cup']