yolov5_reader.yml 582 B

1234567891011121314151617181920212223242526
  1. metric: COCO
  2. num_classes: 80
  3. # Datset configuration
  4. TrainDataset:
  5. !COCODataSet
  6. image_dir: train2017
  7. anno_path: annotations/instances_train2017.json
  8. dataset_dir: dataset/coco/
  9. EvalDataset:
  10. !COCODataSet
  11. image_dir: val2017
  12. anno_path: annotations/instances_val2017.json
  13. dataset_dir: dataset/coco/
  14. worker_num: 0
  15. # preprocess reader in test
  16. TestReader:
  17. sample_transforms:
  18. - Decode: {}
  19. - Resize: {target_size: [640, 640], keep_ratio: True, interp: 1}
  20. - Pad: {size: [640, 640], fill_value: [114., 114., 114.]}
  21. - Permute: {}
  22. batch_size: 1