forked from PulseFocusPlatform/PulseFocusPlatform
131 lines
2.9 KiB
YAML
131 lines
2.9 KiB
YAML
architecture: BlazeFace
|
|
max_iters: 160000
|
|
pretrain_weights:
|
|
use_gpu: true
|
|
snapshot_iter: 10000
|
|
log_iter: 20
|
|
metric: WIDERFACE
|
|
save_dir: output
|
|
weights: output/blazeface_keypoint/model_final.pdparams
|
|
# 1(label_class) + 1(background)
|
|
num_classes: 2
|
|
with_lmk: true
|
|
|
|
BlazeFace:
|
|
backbone: BlazeNet
|
|
output_decoder:
|
|
keep_top_k: 750
|
|
nms_threshold: 0.3
|
|
nms_top_k: 5000
|
|
score_threshold: 0.01
|
|
min_sizes: [[16.,24.], [32., 48., 64., 80., 96., 128.]]
|
|
use_density_prior_box: false
|
|
lmk_loss:
|
|
overlap_threshold: 0.35
|
|
neg_overlap: 0.35
|
|
|
|
BlazeNet:
|
|
with_extra_blocks: true
|
|
lite_edition: false
|
|
|
|
LearningRate:
|
|
base_lr: 0.002
|
|
schedulers:
|
|
- !PiecewiseDecay
|
|
gamma: 0.1
|
|
milestones: [120000, 150000]
|
|
|
|
OptimizerBuilder:
|
|
optimizer:
|
|
momentum: 0.9
|
|
type: Momentum
|
|
regularizer:
|
|
factor: 0.0005
|
|
type: L2
|
|
|
|
TrainReader:
|
|
inputs_def:
|
|
image_shape: [3, 640, 640]
|
|
fields: ['image', 'gt_bbox', 'gt_class', 'gt_keypoint', 'keypoint_ignore']
|
|
dataset:
|
|
!WIDERFaceDataSet
|
|
dataset_dir: dataset/wider_face
|
|
anno_path: wider_face_split/wider_face_train_bbx_lmk_gt.txt
|
|
image_dir: WIDER_train/images
|
|
with_lmk: true
|
|
sample_transforms:
|
|
- !DecodeImage
|
|
to_rgb: true
|
|
- !NormalizeBox {}
|
|
- !RandomDistort
|
|
brightness_lower: 0.875
|
|
brightness_upper: 1.125
|
|
is_order: true
|
|
- !ExpandImage
|
|
max_ratio: 4
|
|
prob: 0.5
|
|
- !CropImageWithDataAchorSampling
|
|
anchor_sampler:
|
|
- [1, 10, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.2, 0.0]
|
|
batch_sampler:
|
|
- [1, 50, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 1.0, 0.0]
|
|
- [1, 50, 0.3, 1.0, 1.0, 1.0, 0.0, 0.0, 1.0, 0.0]
|
|
- [1, 50, 0.3, 1.0, 1.0, 1.0, 0.0, 0.0, 1.0, 0.0]
|
|
- [1, 50, 0.3, 1.0, 1.0, 1.0, 0.0, 0.0, 1.0, 0.0]
|
|
- [1, 50, 0.3, 1.0, 1.0, 1.0, 0.0, 0.0, 1.0, 0.0]
|
|
target_size: 640
|
|
- !ResizeImage
|
|
target_size: 640
|
|
interp: 1
|
|
- !RandomInterpImage
|
|
target_size: 640
|
|
- !RandomFlipImage
|
|
is_normalized: true
|
|
- !Permute {}
|
|
- !NormalizeImage
|
|
is_scale: false
|
|
mean: [104, 117, 123]
|
|
std: [127.502231, 127.502231, 127.502231]
|
|
batch_size: 16
|
|
use_process: true
|
|
worker_num: 8
|
|
shuffle: true
|
|
|
|
EvalReader:
|
|
inputs_def:
|
|
fields: ['image', 'im_id']
|
|
dataset:
|
|
!WIDERFaceDataSet
|
|
dataset_dir: dataset/wider_face
|
|
anno_path: wider_face_split/wider_face_val_bbx_gt.txt
|
|
image_dir: WIDER_val/images
|
|
sample_transforms:
|
|
- !DecodeImage
|
|
to_rgb: true
|
|
- !NormalizeBox {}
|
|
- !Permute {}
|
|
- !NormalizeImage
|
|
is_scale: false
|
|
mean: [104, 117, 123]
|
|
std: [127.502231, 127.502231, 127.502231]
|
|
batch_size: 1
|
|
|
|
TestReader:
|
|
inputs_def:
|
|
fields: ['image', 'im_id', 'im_shape']
|
|
dataset:
|
|
!ImageFolder
|
|
use_default_label: true
|
|
sample_transforms:
|
|
- !DecodeImage
|
|
to_rgb: true
|
|
- !ResizeImage
|
|
target_size: 640
|
|
interp: 1
|
|
- !Permute {}
|
|
- !NormalizeImage
|
|
is_scale: false
|
|
mean: [104, 117, 123]
|
|
std: [127.502231, 127.502231, 127.502231]
|
|
batch_size: 1
|