wgisd_detection.yml 597 B

1234567891011121314151617181920
  1. metric: COCO
  2. num_classes: 5
  3. TrainDataset:
  4. !COCODataSet
  5. image_dir: data
  6. anno_path: coco_annotations/new_train_bbox_instances.json
  7. dataset_dir: dataset/wgisd/
  8. data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
  9. EvalDataset:
  10. !COCODataSet
  11. image_dir: data
  12. anno_path: coco_annotations/new_test_bbox_instances.json
  13. dataset_dir: dataset/wgisd/
  14. TestDataset:
  15. !ImageFolder
  16. anno_path: coco_annotations/new_test_bbox_instances.json # also support txt (like VOC's label_list.txt)
  17. dataset_dir: dataset/wgisd/ # if set, anno_path will be 'dataset_dir/anno_path'