|
- worker_num: 4
- TrainReader:
- inputs_def:
- num_max_boxes: 50 #输入最多的目标框数
- sample_transforms: #设置各种图像增强方法
- - Decode: {}
- - Mixup: {alpha: 1.5, beta: 1.5}
- - RandomDistort: {}
- - RandomExpand: {fill_value: [123.675, 116.28, 103.53]} #随机扩展图像增强
- - RandomCrop: {} #随机剪裁
- - RandomFlip: {}
- batch_transforms:
- - BatchRandomResize: {target_size: [320, 352, 384, 416, 448, 480, 512, 544, 576, 608], random_size: True, random_interp: True, keep_ratio: False} #将图片压缩到target_size进行输入
- - NormalizeBox: {}
- - PadBox: {num_max_boxes: 50}
- - BboxXYXY2XYWH: {} #标注框标注格式转换(由XYXY格式转换为XYWH格式)
- - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True} #图像归一化
- - Permute: {}
- - Gt2YoloTarget: {anchor_masks: [[6, 7, 8], [3, 4, 5], [0, 1, 2]], anchors: [[10, 13], [16, 30], [33, 23], [30, 61], [62, 45], [59, 119], [116, 90], [156, 198], [373, 326]], downsample_ratios: [32, 16, 8]}
- batch_size: 8 #batchsize大小
- shuffle: true
- drop_last: true #如何处理数据集长度除于batch_size余下的数据,True则舍弃,False则保留
- mixup_epoch: -1 #使用mixup的轮数,-1为不使用mixup
- use_shared_memory: true #是否使用缓冲区
-
- EvalReader:
- inputs_def:
- num_max_boxes: 50
- sample_transforms:
- - Decode: {}
- - Resize: {target_size: [608, 608], keep_ratio: False, interp: 2}
- - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
- - Permute: {}
- batch_size: 1
-
- TestReader:
- inputs_def:
- image_shape: [3, 608, 608] #图像为3通道,尺寸为608*608
- sample_transforms:
- - Decode: {}
- - Resize: {target_size: [608, 608], keep_ratio: False, interp: 2}
- - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
- - Permute: {}
- batch_size: 1
|