yolox_reader.yml 1.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. worker_num: 4
  2. TrainReader:
  3. sample_transforms:
  4. - Decode: {}
  5. - Mosaic:
  6. prob: 1.0
  7. input_dim: [640, 640]
  8. degrees: [-10, 10]
  9. scale: [0.1, 2.0]
  10. shear: [-2, 2]
  11. translate: [-0.1, 0.1]
  12. enable_mixup: True
  13. mixup_prob: 1.0
  14. mixup_scale: [0.5, 1.5]
  15. - AugmentHSV: {is_bgr: False, hgain: 5, sgain: 30, vgain: 30}
  16. - PadResize: {target_size: 640}
  17. - RandomFlip: {}
  18. batch_transforms:
  19. - Permute: {}
  20. batch_size: 8
  21. shuffle: True
  22. drop_last: True
  23. collate_batch: False
  24. mosaic_epoch: 285
  25. EvalReader:
  26. sample_transforms:
  27. - Decode: {}
  28. - Resize: {target_size: [640, 640], keep_ratio: True, interp: 1}
  29. - Pad: {size: [640, 640], fill_value: [114., 114., 114.]}
  30. - Permute: {}
  31. batch_size: 4
  32. TestReader:
  33. inputs_def:
  34. image_shape: [3, 640, 640]
  35. sample_transforms:
  36. - Decode: {}
  37. - Resize: {target_size: [640, 640], keep_ratio: True, interp: 1}
  38. - Pad: {size: [640, 640], fill_value: [114., 114., 114.]}
  39. - Permute: {}
  40. batch_size: 1