coco_detection.yml 590 B

123456789101112131415161718192021
  1. metric: COCO
  2. num_classes: 80
  3. TrainDataset:
  4. !COCODataSet
  5. image_dir: train2017
  6. anno_path: annotations/instances_train2017.json
  7. dataset_dir: dataset/coco
  8. data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
  9. EvalDataset:
  10. !COCODataSet
  11. image_dir: val2017
  12. anno_path: annotations/instances_val2017.json
  13. dataset_dir: dataset/coco
  14. allow_empty: true
  15. TestDataset:
  16. !ImageFolder
  17. anno_path: annotations/instances_val2017.json # also support txt (like VOC's label_list.txt)
  18. dataset_dir: dataset/coco # if set, anno_path will be 'dataset_dir/anno_path'