DocumentAIKit/configs/centernet/_base_/centernet_reader.yml

worker_num: 4
TrainReader:
  inputs_def:
    image_shape: [3, 512, 512]
  sample_transforms:
    - Decode: {}
    - FlipWarpAffine: {keep_res: False, input_h: 512, input_w: 512, use_random: True}
    - CenterRandColor: {}
    - Lighting: {eigval: [0.2141788, 0.01817699, 0.00341571], eigvec: [[-0.58752847, -0.69563484, 0.41340352], [-0.5832747, 0.00994535, -0.81221408], [-0.56089297, 0.71832671, 0.41158938]]}
    - NormalizeImage: {mean: [0.40789655, 0.44719303, 0.47026116], std: [0.2886383 , 0.27408165, 0.27809834], is_scale: False}
    - Permute: {}
    - Gt2CenterNetTarget: {down_ratio: 4, max_objs: 128}
  batch_size: 16
  shuffle: True
  drop_last: True
  use_shared_memory: True

EvalReader:
  sample_transforms:
    - Decode: {}
    - WarpAffine: {keep_res: True, input_h: 512, input_w: 512}
    - NormalizeImage: {mean: [0.40789655, 0.44719303, 0.47026116], std: [0.2886383 , 0.27408165, 0.27809834]}
    - Permute: {}
  batch_size: 1


TestReader:
  inputs_def:
    image_shape: [3, 512, 512]
  sample_transforms:
    - Decode: {}
    - WarpAffine: {keep_res: True, input_h: 512, input_w: 512}
    - NormalizeImage: {mean: [0.40789655, 0.44719303, 0.47026116], std: [0.2886383 , 0.27408165, 0.27809834], is_scale: True}
    - Permute: {}
  batch_size: 1