forked from PulseFocusPlatform/PulseFocusPlatform
82 lines
1.9 KiB
YAML
82 lines
1.9 KiB
YAML
|
_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
|