ssd_reader.yml 1.1 KB
Newer Older
Q
qingqing01 已提交
1 2 3 4 5 6
worker_num: 2
TrainReader:
  inputs_def:
    num_max_boxes: 90

  sample_transforms:
W
wangxinxin08 已提交
7 8 9 10 11 12 13 14
    - Decode: {}
    - RandomDistort: {brightness: [0.5, 1.125, 0.875], random_apply: False}
    - RandomExpand: {fill_value: [104., 117., 123.]}
    - RandomCrop: {allow_no_crop: true}
    - RandomFlip: {}
    - Resize: {target_size: [300, 300], keep_ratio: False, interp: 1}
    - NormalizeBox: {}
    - PadBox: {num_max_boxes: 90}
Q
qingqing01 已提交
15 16

  batch_transforms:
W
wangxinxin08 已提交
17 18
    - NormalizeImage: {mean: [104., 117., 123.], std: [1., 1., 1.], is_scale: false}
    - Permute: {}
Q
qingqing01 已提交
19 20 21 22 23 24 25 26

  batch_size: 8
  shuffle: true
  drop_last: true


EvalReader:
  sample_transforms:
W
wangxinxin08 已提交
27 28 29 30
    - Decode: {}
    - Resize: {target_size: [300, 300], keep_ratio: False, interp: 1}
    - NormalizeImage: {mean: [104., 117., 123.], std: [1., 1., 1.], is_scale: false}
    - Permute: {}
Q
qingqing01 已提交
31 32 33 34 35 36 37
  batch_size: 1
  drop_empty: false

TestReader:
  inputs_def:
    image_shape: [3, 300, 300]
  sample_transforms:
W
wangxinxin08 已提交
38 39 40 41
    - Decode: {}
    - Resize: {target_size: [300, 300], keep_ratio: False, interp: 1}
    - NormalizeImage: {mean: [104., 117., 123.], std: [1., 1., 1.], is_scale: false}
    - Permute: {}
Q
qingqing01 已提交
42
  batch_size: 1