faster_reader.yml 1.4 KB
Newer Older
1
worker_num: 2
F
FDInSky 已提交
2 3 4 5
TrainReader:
  inputs_def:
    fields: ['image', 'im_info', 'im_id', 'gt_bbox', 'gt_class', 'is_crowd']
  sample_transforms:
6 7 8 9 10
  - DecodeImage: {to_rgb: true}
  - RandomFlipImage: {prob: 0.5}
  - NormalizeImage: {is_channel_first: false, is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]}
  - ResizeImage: {target_size: 800, max_size: 1333, interp: 1, use_cv2: true}
  - Permute: {to_bgr: false, channel_first: true}
F
FDInSky 已提交
11
  batch_transforms:
12
  - PadBatch: {pad_to_stride: -1, use_padded_im_info: false, pad_gt: true}
F
FDInSky 已提交
13 14
  batch_size: 1
  shuffle: true
15 16
  drop_last: true

F
FDInSky 已提交
17 18 19

EvalReader:
  inputs_def:
20
    fields: ['image', 'im_shape', 'scale_factor', 'im_id']
F
FDInSky 已提交
21
  sample_transforms:
22 23 24 25
  - DecodeOp: { }
  - NormalizeImageOp: { is_scale: true, mean: [ 0.485,0.456,0.406 ], std: [ 0.229, 0.224,0.225 ] }
  - ResizeOp: { interp: 1, target_size: [ 800, 1333 ] }
  - PermuteOp: { }
F
FDInSky 已提交
26
  batch_transforms:
27 28
  - PadBatchOp: { pad_to_stride: -1, pad_gt: false }
  batch_size: 1
F
FDInSky 已提交
29
  shuffle: false
30
  drop_last: false
F
FDInSky 已提交
31
  drop_empty: false
32

F
FDInSky 已提交
33 34 35

TestReader:
  inputs_def:
36
    fields: ['image', 'im_shape', 'scale_factor', 'im_id']
F
FDInSky 已提交
37
  sample_transforms:
38 39 40 41 42 43
  - DecodeOp: { }
  - NormalizeImageOp: { is_scale: true, mean: [ 0.485,0.456,0.406 ], std: [ 0.229, 0.224,0.225 ] }
  - ResizeOp: { interp: 1, target_size: [ 800, 1333 ] }
  - PermuteOp: { }
  batch_transforms:
  - PadBatchOp: { pad_to_stride: -1, pad_gt: false }
F
FDInSky 已提交
44 45
  batch_size: 1
  shuffle: false
46
  drop_last: false