coco_detection_percent_1.yml 963 B

12345678910111213141516171819202122232425262728293031
  1. metric: COCO
  2. num_classes: 80
  3. # partial labeled COCO, use `SemiCOCODataSet` rather than `COCODataSet`
  4. TrainDataset:
  5. !SemiCOCODataSet
  6. image_dir: train2017
  7. anno_path: semi_annotations/instances_train2017.1@1.json
  8. dataset_dir: dataset/coco
  9. data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
  10. # partial unlabeled COCO, use `SemiCOCODataSet` rather than `COCODataSet`
  11. UnsupTrainDataset:
  12. !SemiCOCODataSet
  13. image_dir: train2017
  14. anno_path: semi_annotations/instances_train2017.1@1-unlabeled.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'