22 lines
528 B
YAML
22 lines
528 B
YAML
metric: RBOX
|
|
num_classes: 9
|
|
|
|
TrainDataset:
|
|
!COCODataSet
|
|
image_dir: images
|
|
anno_path: annotations/train.json
|
|
dataset_dir: dataset/spine_coco
|
|
data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd', 'gt_poly']
|
|
|
|
EvalDataset:
|
|
!COCODataSet
|
|
image_dir: images
|
|
anno_path: annotations/valid.json
|
|
dataset_dir: dataset/spine_coco
|
|
data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd', 'gt_poly']
|
|
|
|
TestDataset:
|
|
!ImageFolder
|
|
anno_path: annotations/valid.json
|
|
dataset_dir: dataset/spine_coco
|