spine_coco.yml 455 B

1234567891011121314151617181920
  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_rbox']
  9. EvalDataset:
  10. !COCODataSet
  11. image_dir: images
  12. anno_path: annotations/valid.json
  13. dataset_dir: dataset/spine_coco
  14. TestDataset:
  15. !ImageFolder
  16. anno_path: annotations/valid.json
  17. dataset_dir: dataset/spine_coco