jde_reader_576x320.yml 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. worker_num: 2
  2. TrainReader:
  3. sample_transforms:
  4. - Decode: {}
  5. - RGBReverse: {}
  6. - AugmentHSV: {}
  7. - LetterBoxResize: {target_size: [320, 576]}
  8. - MOTRandomAffine: {}
  9. - RandomFlip: {}
  10. - BboxXYXY2XYWH: {}
  11. - NormalizeBox: {}
  12. - NormalizeImage: {mean: [0, 0, 0], std: [1, 1, 1], is_scale: True}
  13. - RGBReverse: {}
  14. - Permute: {}
  15. batch_transforms:
  16. - Gt2JDETargetThres:
  17. anchor_masks: [[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11]]
  18. anchors: [[[85,255], [120,360], [170,420], [340,420]],
  19. [[21,64], [30,90], [43,128], [60,180]],
  20. [[6,16], [8,23], [11,32], [16,45]]]
  21. downsample_ratios: [32, 16, 8]
  22. ide_thresh: 0.5
  23. fg_thresh: 0.5
  24. bg_thresh: 0.4
  25. batch_size: 4
  26. shuffle: true
  27. drop_last: true
  28. use_shared_memory: true
  29. EvalMOTReader:
  30. sample_transforms:
  31. - Decode: {}
  32. - LetterBoxResize: {target_size: [320, 576]}
  33. - NormalizeImage: {mean: [0, 0, 0], std: [1, 1, 1], is_scale: True}
  34. - Permute: {}
  35. batch_size: 1
  36. TestMOTReader:
  37. inputs_def:
  38. image_shape: [3, 320, 576]
  39. sample_transforms:
  40. - Decode: {}
  41. - LetterBoxResize: {target_size: [320, 576]}
  42. - NormalizeImage: {mean: [0, 0, 0], std: [1, 1, 1], is_scale: True}
  43. - Permute: {}
  44. batch_size: 1