coco_detection.yml 580 Bytes
Newer Older
panhb's avatar
panhb committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
metric: COCO
num_classes: 80

TrainDataset:
  name: COCODataSet
  image_dir: train2017
  anno_path: annotations/instances_train2017.json
  dataset_dir: /dataset/COCO2017
  data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']

EvalDataset:
  name: COCODataSet
  image_dir: val2017
  anno_path: annotations/instances_val2017.json
  dataset_dir: /dataset/COCO2017

TestDataset:
  name: ImageFolder
  anno_path: annotations/instances_val2017.json # also support txt (like VOC's label_list.txt)
  dataset_dir: /dataset/COCO2017 # if set, anno_path will be 'dataset_dir/anno_path'