ssdlite320_reader.yml 1.1 KB
Newer Older
1 2 3 4 5
worker_num: 8
TrainReader:
  inputs_def:
    num_max_boxes: 90
  sample_transforms:
W
wangxinxin08 已提交
6 7 8 9 10 11 12 13
    - Decode: {}
    - RandomDistort: {brightness: [0.5, 1.125, 0.875], random_apply: False}
    - RandomExpand: {fill_value: [123.675, 116.28, 103.53]}
    - RandomCrop: {allow_no_crop: Fasle}
    - RandomFlip: {}
    - Resize: {target_size: [320, 320], keep_ratio: False, interp: 1}
    - NormalizeBox: {}
    - PadBox: {num_max_boxes: 90}
14
  batch_transforms:
W
wangxinxin08 已提交
15 16
    - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: true}
    - Permute: {}
17 18 19 20 21 22 23
  batch_size: 64
  shuffle: true
  drop_last: true


EvalReader:
  sample_transforms:
W
wangxinxin08 已提交
24 25 26 27
    - Decode: {}
    - Resize: {target_size: [320, 320], keep_ratio: False, interp: 1}
    - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: true}
    - Permute: {}
28 29 30 31 32 33 34
  batch_size: 1


TestReader:
  inputs_def:
    image_shape: [3, 320, 320]
  sample_transforms:
W
wangxinxin08 已提交
35 36 37 38
    - Decode: {}
    - Resize: {target_size: [320, 320], keep_ratio: False, interp: 1}
    - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: true}
    - Permute: {}
39
  batch_size: 1