DocumentAIKit/configs/ssd/_base_/ssd_r34_reader.yml

worker_num: 3
TrainReader:
  inputs_def:
    num_max_boxes: 90
  sample_transforms:
    - Decode: {}
    - RandomCrop: {num_attempts: 1}
    - RandomFlip: {}
    - Resize: {target_size: [300, 300], keep_ratio: False, interp: 1}
    - RandomDistort: {brightness: [0.875, 1.125, 0.5], random_apply: False}
    - NormalizeBox: {}
    - PadBox: {num_max_boxes: 90}
    - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: true}
    - Permute: {}
  batch_size: 64
  shuffle: true
  drop_last: true
  use_shared_memory: true


EvalReader:
  sample_transforms:
    - Decode: {}
    - Resize: {target_size: [300, 300], keep_ratio: False, interp: 1}
    - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: true}
    - Permute: {}
  batch_size: 1


TestReader:
  inputs_def:
    image_shape: [3, 300, 300]
  sample_transforms:
    - Decode: {}
    - Resize: {target_size: [300, 300], keep_ratio: False, interp: 1}
    - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: true}
    - Permute: {}
  batch_size: 1