44 lines
1.2 KiB
YAML
44 lines
1.2 KiB
YAML
worker_num: 4
|
|
TrainReader:
|
|
sample_transforms:
|
|
- Decode: {}
|
|
- RandomDistort: {prob: 0.8}
|
|
- RandomExpand: {fill_value: [123.675, 116.28, 103.53]}
|
|
- RandomCrop: {prob: 0.8}
|
|
- RandomFlip: {}
|
|
batch_transforms:
|
|
- BatchRandomResize: {target_size: [480, 512, 544, 576, 608, 640, 640, 640, 672, 704, 736, 768, 800], random_size: True, random_interp: True, keep_ratio: False}
|
|
- NormalizeImage: {mean: [0., 0., 0.], std: [1., 1., 1.], norm_type: none}
|
|
- NormalizeBox: {}
|
|
- BboxXYXY2XYWH: {}
|
|
- Permute: {}
|
|
batch_size: 4
|
|
shuffle: true
|
|
drop_last: true
|
|
collate_batch: false
|
|
use_shared_memory: false
|
|
|
|
|
|
EvalReader:
|
|
sample_transforms:
|
|
- Decode: {}
|
|
- Resize: {target_size: [640, 640], keep_ratio: False, interp: 2} # target_size: (h, w)
|
|
- NormalizeImage: {mean: [0., 0., 0.], std: [1., 1., 1.], norm_type: none}
|
|
- Permute: {}
|
|
batch_size: 4
|
|
shuffle: false
|
|
drop_last: false
|
|
|
|
|
|
TestReader:
|
|
inputs_def:
|
|
image_shape: [3, 640, 640]
|
|
sample_transforms:
|
|
- Decode: {}
|
|
- Resize: {target_size: [640, 640], keep_ratio: False, interp: 2}
|
|
- NormalizeImage: {mean: [0., 0., 0.], std: [1., 1., 1.], norm_type: none}
|
|
- Permute: {}
|
|
batch_size: 1
|
|
shuffle: false
|
|
drop_last: false
|