spine_coco.yml 528 B

123456789101112131415161718192021
  1. metric: RBOX
  2. num_classes: 9
  3. TrainDataset:
  4. !COCODataSet
  5. image_dir: images
  6. anno_path: annotations/train.json
  7. dataset_dir: dataset/spine_coco
  8. data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd', 'gt_poly']
  9. EvalDataset:
  10. !COCODataSet
  11. image_dir: images
  12. anno_path: annotations/valid.json
  13. dataset_dir: dataset/spine_coco
  14. data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd', 'gt_poly']
  15. TestDataset:
  16. !ImageFolder
  17. anno_path: annotations/valid.json
  18. dataset_dir: dataset/spine_coco