Skip to content

Commit 8fbdf1c

Browse files
authored
add ppyoloe plus cfgs (#6686)
1 parent 0f75cb6 commit 8fbdf1c

File tree

5 files changed

+160
-0
lines changed

5 files changed

+160
-0
lines changed
Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
2+
Global:
3+
reader_config: configs/ppyoloe_plus_reader.yml
4+
input_list: ['image', 'scale_factor']
5+
arch: YOLO
6+
Evaluation: True
7+
model_dir: ./ppyoloe_plus_crn_l_80e_coco
8+
model_filename: model.pdmodel
9+
params_filename: model.pdiparams
10+
11+
Distillation:
12+
alpha: 1.0
13+
loss: soft_label
14+
15+
Quantization:
16+
use_pact: true
17+
activation_quantize_type: 'moving_average_abs_max'
18+
quantize_op_types:
19+
- conv2d
20+
- depthwise_conv2d
21+
22+
TrainConfig:
23+
train_iter: 5000
24+
eval_iter: 1000
25+
learning_rate:
26+
type: CosineAnnealingDecay
27+
learning_rate: 0.00003
28+
T_max: 6000
29+
optimizer_builder:
30+
optimizer:
31+
type: SGD
32+
weight_decay: 4.0e-05
33+
Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
2+
Global:
3+
reader_config: configs/ppyoloe_plus_reader.yml
4+
input_list: ['image', 'scale_factor']
5+
arch: YOLO
6+
Evaluation: True
7+
model_dir: ./ppyoloe_plus_crn_m_80e_coco
8+
model_filename: model.pdmodel
9+
params_filename: model.pdiparams
10+
11+
Distillation:
12+
alpha: 1.0
13+
loss: soft_label
14+
15+
Quantization:
16+
use_pact: true
17+
activation_quantize_type: 'moving_average_abs_max'
18+
quantize_op_types:
19+
- conv2d
20+
- depthwise_conv2d
21+
22+
TrainConfig:
23+
train_iter: 5000
24+
eval_iter: 1000
25+
learning_rate:
26+
type: CosineAnnealingDecay
27+
learning_rate: 0.00003
28+
T_max: 6000
29+
optimizer_builder:
30+
optimizer:
31+
type: SGD
32+
weight_decay: 4.0e-05
33+
Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
2+
3+
metric: COCO
4+
num_classes: 80
5+
6+
# Datset configuration
7+
TrainDataset:
8+
!COCODataSet
9+
image_dir: train2017
10+
anno_path: annotations/instances_train2017.json
11+
dataset_dir: dataset/coco/
12+
13+
EvalDataset:
14+
!COCODataSet
15+
image_dir: val2017
16+
anno_path: annotations/instances_val2017.json
17+
dataset_dir: dataset/coco/
18+
19+
worker_num: 0
20+
21+
# preprocess reader in test
22+
EvalReader:
23+
sample_transforms:
24+
- Decode: {}
25+
- Resize: {target_size: [640, 640], keep_ratio: False, interp: 2}
26+
- NormalizeImage: {mean: [0., 0., 0.], std: [1., 1., 1.], is_scale: True}
27+
- Permute: {}
28+
batch_size: 4
Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
2+
Global:
3+
reader_config: configs/ppyoloe_plus_reader.yml
4+
input_list: ['image', 'scale_factor']
5+
arch: YOLO
6+
Evaluation: True
7+
model_dir: ./ppyoloe_plus_crn_s_80e_coco
8+
model_filename: model.pdmodel
9+
params_filename: model.pdiparams
10+
11+
Distillation:
12+
alpha: 1.0
13+
loss: soft_label
14+
15+
Quantization:
16+
use_pact: true
17+
activation_quantize_type: 'moving_average_abs_max'
18+
quantize_op_types:
19+
- conv2d
20+
- depthwise_conv2d
21+
22+
TrainConfig:
23+
train_iter: 5000
24+
eval_iter: 1000
25+
learning_rate:
26+
type: CosineAnnealingDecay
27+
learning_rate: 0.00003
28+
T_max: 6000
29+
optimizer_builder:
30+
optimizer:
31+
type: SGD
32+
weight_decay: 4.0e-05
33+
Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
2+
Global:
3+
reader_config: configs/ppyoloe_plus_reader.yml
4+
input_list: ['image', 'scale_factor']
5+
arch: YOLO
6+
Evaluation: True
7+
model_dir: ./ppyoloe_plus_crn_x_80e_coco
8+
model_filename: model.pdmodel
9+
params_filename: model.pdiparams
10+
11+
Distillation:
12+
alpha: 1.0
13+
loss: soft_label
14+
15+
Quantization:
16+
use_pact: true
17+
activation_quantize_type: 'moving_average_abs_max'
18+
quantize_op_types:
19+
- conv2d
20+
- depthwise_conv2d
21+
22+
TrainConfig:
23+
train_iter: 5000
24+
eval_iter: 1000
25+
learning_rate:
26+
type: CosineAnnealingDecay
27+
learning_rate: 0.00003
28+
T_max: 6000
29+
optimizer_builder:
30+
optimizer:
31+
type: SGD
32+
weight_decay: 4.0e-05
33+

0 commit comments

Comments
 (0)