| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081 |
- _BASE_: [
- '../datasets/coco_detection.yml',
- '../runtime.yml',
- './_base_/ppyolo_r18vd.yml',
- './_base_/optimizer_1x.yml',
- './_base_/ppyolo_reader.yml',
- ]
- snapshot_epoch: 10
- weights: output/ppyolo_r18vd_coco/model_final
- TrainReader:
- 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
- - NormalizeBox: {}
- - PadBox: {num_max_boxes: 50}
- - BboxXYXY2XYWH: {}
- - NormalizeImage:
- mean: [0.485, 0.456, 0.406]
- std: [0.229, 0.224, 0.225]
- is_scale: True
- - Permute: {}
- - Gt2YoloTarget:
- anchor_masks: [[3, 4, 5], [0, 1, 2]]
- anchors: [[10, 14], [23, 27], [37, 58], [81, 82], [135, 169], [344, 319]]
- downsample_ratios: [32, 16]
- batch_size: 32
- mixup_epoch: 500
- shuffle: true
- EvalReader:
- sample_transforms:
- - Decode: {}
- - Resize: {target_size: [512, 512], 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: 8
- TestReader:
- inputs_def:
- image_shape: [3, 512, 512]
- sample_transforms:
- - Decode: {}
- - Resize: {target_size: [512, 512], 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
- epoch: 270
- LearningRate:
- base_lr: 0.004
- schedulers:
- - !PiecewiseDecay
- gamma: 0.1
- milestones:
- - 162
- - 216
- - !LinearWarmup
- start_factor: 0.
- steps: 4000
- OptimizerBuilder:
- optimizer:
- momentum: 0.9
- type: Momentum
- regularizer:
- factor: 0.0005
- type: L2
|
