ssd_mobilenet_reader.yml 1.2 KB
Newer Older
1 2 3 4 5 6 7 8 9 10
worker_num: 8
TrainReader:
  inputs_def:
    num_max_boxes: 90
  sample_transforms:
    - DecodeOp: {}
    - RandomDistortOp: {brightness: [0.5, 1.125, 0.875], random_apply: False}
    - RandomExpandOp: {fill_value: [127.5, 127.5, 127.5]}
    - RandomCropOp: {allow_no_crop: Fasle}
    - RandomFlipOp: {}
F
Feng Ni 已提交
11
    - ResizeOp: {target_size: [300, 300], keep_ratio: False, interp: 1}
12 13 14 15
    - NormalizeBoxOp: {}
    - PadBoxOp: {num_max_boxes: 90}
  batch_transforms:
    - NormalizeImageOp: {mean: [127.5, 127.5, 127.5], std: [127.502231, 127.502231, 127.502231], is_scale: false}
F
Feng Ni 已提交
16
    - Permute: {}
17 18 19 20 21 22 23 24 25 26
  batch_size: 32
  shuffle: true
  drop_last: true


EvalReader:
  sample_transforms:
    - DecodeOp: {}
    - ResizeOp: {target_size: [300, 300], keep_ratio: False, interp: 1}
    - NormalizeImageOp: {mean: [127.5, 127.5, 127.5], std: [127.502231, 127.502231, 127.502231], is_scale: false}
F
Feng Ni 已提交
27
    - Permute: {}
28 29 30 31 32 33 34 35 36 37 38
  batch_size: 1
  drop_empty: false


TestReader:
  inputs_def:
    image_shape: [3, 300, 300]
  sample_transforms:
    - DecodeOp: {}
    - ResizeOp: {target_size: [300, 300], keep_ratio: False, interp: 1}
    - NormalizeImageOp: {mean: [127.5, 127.5, 127.5], std: [127.502231, 127.502231, 127.502231], is_scale: false}
F
Feng Ni 已提交
39
    - Permute: {}
40
  batch_size: 1