coco_detection_full.yml 937 B

12345678910111213141516171819202122232425262728293031
  1. metric: COCO
  2. num_classes: 80
  3. # full labeled COCO, use `SemiCOCODataSet` rather than `COCODataSet`
  4. TrainDataset:
  5. !SemiCOCODataSet
  6. image_dir: train2017
  7. anno_path: annotations/instances_train2017.json
  8. dataset_dir: dataset/coco
  9. data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
  10. # full unlabeled COCO, use `SemiCOCODataSet` rather than `COCODataSet`
  11. UnsupTrainDataset:
  12. !SemiCOCODataSet
  13. image_dir: unlabeled2017
  14. anno_path: annotations/instances_unlabeled2017.json
  15. dataset_dir: dataset/coco
  16. data_fields: ['image']
  17. supervised: False
  18. EvalDataset:
  19. !COCODataSet
  20. image_dir: val2017
  21. anno_path: annotations/instances_val2017.json
  22. dataset_dir: dataset/coco
  23. allow_empty: true
  24. TestDataset:
  25. !ImageFolder
  26. anno_path: annotations/instances_val2017.json # also support txt (like VOC's label_list.txt)
  27. dataset_dir: dataset/coco # if set, anno_path will be 'dataset_dir/anno_path'