| task: detection | |
| evaluator: | |
| type: CocoEvaluator | |
| iou_types: ['bbox', ] | |
| num_classes: 4 # your dataset classes | |
| remap_mscoco_category: False | |
| train_dataloader: | |
| type: DataLoader | |
| dataset: | |
| type: CocoDetection | |
| img_folder: ./datasets/train | |
| ann_file: ./datasets/train/_annotations.coco.json | |
| return_masks: False | |
| transforms: | |
| type: Compose | |
| ops: ~ | |
| shuffle: True | |
| num_workers: 4 | |
| drop_last: True | |
| collate_fn: | |
| type: BatchImageCollateFunction | |
| val_dataloader: | |
| type: DataLoader | |
| dataset: | |
| type: CocoDetection | |
| img_folder: ./datasets/valid | |
| ann_file: ./datasets/valid/_annotations.coco.json | |
| return_masks: False | |
| transforms: | |
| type: Compose | |
| ops: ~ | |
| shuffle: False | |
| num_workers: 4 | |
| drop_last: False | |
| collate_fn: | |
| type: BatchImageCollateFunction | |