picodet_reader.yml 737 B

1234567891011121314151617181920212223242526272829303132
  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: 6
  15. eval_height: &eval_height 416
  16. eval_width: &eval_width 416
  17. eval_size: &eval_size [*eval_height, *eval_width]
  18. EvalReader:
  19. sample_transforms:
  20. - Decode: {}
  21. - Resize: {interp: 2, target_size: *eval_size, keep_ratio: False}
  22. - NormalizeImage: {mean: [0, 0, 0], std: [1, 1, 1], is_scale: True}
  23. - Permute: {}
  24. batch_transforms:
  25. - PadBatch: {pad_to_stride: 32}
  26. batch_size: 8
  27. shuffle: false