From 29356b0743e8cfce45f8a4a6428d01bb84ce3512 Mon Sep 17 00:00:00 2001 From: Wenyu Date: Thu, 11 Aug 2022 16:41:43 +0800 Subject: [PATCH] add yolo series auto compress configs (#6623) * add yolo series auto compress configs, test=document_fix * update name, test=document_fix * update yolox reader * update model reader --- .../configs/ppyoloe_l_qat_dis.yaml | 2 +- .../{yolo_reader.yml => ppyoloe_reader.yml} | 0 .../configs/yolov5_reader.yml | 26 +++++++++++++++ .../configs/yolov5_s_qat_dis.yml | 32 ++++++++++++++++++ .../configs/yolov6mt_reader.yml | 26 +++++++++++++++ .../configs/yolov6mt_s_qat_dis.yaml | 33 +++++++++++++++++++ .../configs/yolov7_l_qat_dis.yaml | 33 +++++++++++++++++++ .../configs/yolov7_reader.yml | 26 +++++++++++++++ .../auto_compression/configs/yolox_reader.yml | 26 +++++++++++++++ .../configs/yolox_s_qat_dis.yaml | 33 +++++++++++++++++++ 10 files changed, 236 insertions(+), 1 deletion(-) rename deploy/auto_compression/configs/{yolo_reader.yml => ppyoloe_reader.yml} (100%) create mode 100644 deploy/auto_compression/configs/yolov5_reader.yml create mode 100644 deploy/auto_compression/configs/yolov5_s_qat_dis.yml create mode 100644 deploy/auto_compression/configs/yolov6mt_reader.yml create mode 100644 deploy/auto_compression/configs/yolov6mt_s_qat_dis.yaml create mode 100644 deploy/auto_compression/configs/yolov7_l_qat_dis.yaml create mode 100644 deploy/auto_compression/configs/yolov7_reader.yml create mode 100644 deploy/auto_compression/configs/yolox_reader.yml create mode 100644 deploy/auto_compression/configs/yolox_s_qat_dis.yaml diff --git a/deploy/auto_compression/configs/ppyoloe_l_qat_dis.yaml b/deploy/auto_compression/configs/ppyoloe_l_qat_dis.yaml index cd39981c1..4c7b684ea 100644 --- a/deploy/auto_compression/configs/ppyoloe_l_qat_dis.yaml +++ b/deploy/auto_compression/configs/ppyoloe_l_qat_dis.yaml @@ -1,6 +1,6 @@ Global: - reader_config: configs/yolo_reader.yml + reader_config: configs/ppyoloe_reader.yml input_list: ['image', 'scale_factor'] arch: YOLO Evaluation: True diff --git a/deploy/auto_compression/configs/yolo_reader.yml b/deploy/auto_compression/configs/ppyoloe_reader.yml similarity index 100% rename from deploy/auto_compression/configs/yolo_reader.yml rename to deploy/auto_compression/configs/ppyoloe_reader.yml diff --git a/deploy/auto_compression/configs/yolov5_reader.yml b/deploy/auto_compression/configs/yolov5_reader.yml new file mode 100644 index 000000000..6ad321a04 --- /dev/null +++ b/deploy/auto_compression/configs/yolov5_reader.yml @@ -0,0 +1,26 @@ +metric: COCO +num_classes: 80 + +# Datset configuration +TrainDataset: + !COCODataSet + image_dir: train2017 + anno_path: annotations/instances_train2017.json + dataset_dir: dataset/coco/ + +EvalDataset: + !COCODataSet + image_dir: val2017 + anno_path: annotations/instances_val2017.json + dataset_dir: dataset/coco/ + +worker_num: 0 + +# preprocess reader in test +TestReader: + sample_transforms: + - Decode: {} + - Resize: {target_size: [640, 640], keep_ratio: True, interp: 1} + - Pad: {size: [640, 640], fill_value: [114., 114., 114.]} + - Permute: {} + batch_size: 1 diff --git a/deploy/auto_compression/configs/yolov5_s_qat_dis.yml b/deploy/auto_compression/configs/yolov5_s_qat_dis.yml new file mode 100644 index 000000000..60b98d948 --- /dev/null +++ b/deploy/auto_compression/configs/yolov5_s_qat_dis.yml @@ -0,0 +1,32 @@ + +Global: + reader_config: configs/yolov5_reader.yml + input_list: ['image', 'scale_factor'] + arch: YOLO + Evaluation: True + model_dir: ./yolov5_s_300e_coco + model_filename: model.pdmodel + params_filename: model.pdiparams + +Distillation: + alpha: 1.0 + loss: soft_label + +Quantization: + use_pact: true + activation_quantize_type: 'moving_average_abs_max' + quantize_op_types: + - conv2d + - depthwise_conv2d + +TrainConfig: + train_iter: 5000 + eval_iter: 1000 + learning_rate: + type: CosineAnnealingDecay + learning_rate: 0.00003 + T_max: 6000 + optimizer_builder: + optimizer: + type: SGD + weight_decay: 4.0e-05 diff --git a/deploy/auto_compression/configs/yolov6mt_reader.yml b/deploy/auto_compression/configs/yolov6mt_reader.yml new file mode 100644 index 000000000..9832845b2 --- /dev/null +++ b/deploy/auto_compression/configs/yolov6mt_reader.yml @@ -0,0 +1,26 @@ +metric: COCO +num_classes: 80 + +# Datset configuration +TrainDataset: + !COCODataSet + image_dir: train2017 + anno_path: annotations/instances_train2017.json + dataset_dir: dataset/coco/ + +EvalDataset: + !COCODataSet + image_dir: val2017 + anno_path: annotations/instances_val2017.json + dataset_dir: dataset/coco/ + +worker_num: 0 + +# preprocess reader in test +EvalReader: + sample_transforms: + - Decode: {} + - Resize: {target_size: [640, 640], keep_ratio: True, interp: 1} + - Pad: {size: [640, 640], fill_value: [114., 114., 114.]} + - Permute: {} + batch_size: 1 diff --git a/deploy/auto_compression/configs/yolov6mt_s_qat_dis.yaml b/deploy/auto_compression/configs/yolov6mt_s_qat_dis.yaml new file mode 100644 index 000000000..12ab5b87d --- /dev/null +++ b/deploy/auto_compression/configs/yolov6mt_s_qat_dis.yaml @@ -0,0 +1,33 @@ + +Global: + reader_config: configs/yolov6mt_reader.yml + input_list: ['image', 'scale_factor'] + arch: YOLO + Evaluation: True + model_dir: ./yolov6mt_s_400e_coco + model_filename: model.pdmodel + params_filename: model.pdiparams + +Distillation: + alpha: 1.0 + loss: soft_label + +Quantization: + use_pact: true + activation_quantize_type: 'moving_average_abs_max' + quantize_op_types: + - conv2d + - depthwise_conv2d + +TrainConfig: + train_iter: 5000 + eval_iter: 1000 + learning_rate: + type: CosineAnnealingDecay + learning_rate: 0.00003 + T_max: 6000 + optimizer_builder: + optimizer: + type: SGD + weight_decay: 4.0e-05 + diff --git a/deploy/auto_compression/configs/yolov7_l_qat_dis.yaml b/deploy/auto_compression/configs/yolov7_l_qat_dis.yaml new file mode 100644 index 000000000..408dfd47b --- /dev/null +++ b/deploy/auto_compression/configs/yolov7_l_qat_dis.yaml @@ -0,0 +1,33 @@ + +Global: + reader_config: configs/yolov7_reader.yml + input_list: ['image', 'scale_factor'] + arch: YOLO + Evaluation: True + model_dir: ./yolov7_l_300e_coco + model_filename: model.pdmodel + params_filename: model.pdiparams + +Distillation: + alpha: 1.0 + loss: soft_label + +Quantization: + use_pact: true + activation_quantize_type: 'moving_average_abs_max' + quantize_op_types: + - conv2d + - depthwise_conv2d + +TrainConfig: + train_iter: 5000 + eval_iter: 1000 + learning_rate: + type: CosineAnnealingDecay + learning_rate: 0.00003 + T_max: 6000 + optimizer_builder: + optimizer: + type: SGD + weight_decay: 4.0e-05 + diff --git a/deploy/auto_compression/configs/yolov7_reader.yml b/deploy/auto_compression/configs/yolov7_reader.yml new file mode 100644 index 000000000..9832845b2 --- /dev/null +++ b/deploy/auto_compression/configs/yolov7_reader.yml @@ -0,0 +1,26 @@ +metric: COCO +num_classes: 80 + +# Datset configuration +TrainDataset: + !COCODataSet + image_dir: train2017 + anno_path: annotations/instances_train2017.json + dataset_dir: dataset/coco/ + +EvalDataset: + !COCODataSet + image_dir: val2017 + anno_path: annotations/instances_val2017.json + dataset_dir: dataset/coco/ + +worker_num: 0 + +# preprocess reader in test +EvalReader: + sample_transforms: + - Decode: {} + - Resize: {target_size: [640, 640], keep_ratio: True, interp: 1} + - Pad: {size: [640, 640], fill_value: [114., 114., 114.]} + - Permute: {} + batch_size: 1 diff --git a/deploy/auto_compression/configs/yolox_reader.yml b/deploy/auto_compression/configs/yolox_reader.yml new file mode 100644 index 000000000..6ad321a04 --- /dev/null +++ b/deploy/auto_compression/configs/yolox_reader.yml @@ -0,0 +1,26 @@ +metric: COCO +num_classes: 80 + +# Datset configuration +TrainDataset: + !COCODataSet + image_dir: train2017 + anno_path: annotations/instances_train2017.json + dataset_dir: dataset/coco/ + +EvalDataset: + !COCODataSet + image_dir: val2017 + anno_path: annotations/instances_val2017.json + dataset_dir: dataset/coco/ + +worker_num: 0 + +# preprocess reader in test +TestReader: + sample_transforms: + - Decode: {} + - Resize: {target_size: [640, 640], keep_ratio: True, interp: 1} + - Pad: {size: [640, 640], fill_value: [114., 114., 114.]} + - Permute: {} + batch_size: 1 diff --git a/deploy/auto_compression/configs/yolox_s_qat_dis.yaml b/deploy/auto_compression/configs/yolox_s_qat_dis.yaml new file mode 100644 index 000000000..6f5d97194 --- /dev/null +++ b/deploy/auto_compression/configs/yolox_s_qat_dis.yaml @@ -0,0 +1,33 @@ + +Global: + reader_config: configs/yolox_reader.yml + input_list: ['image', 'scale_factor'] + arch: YOLO + Evaluation: True + model_dir: ./yolox_s_300e_coco + model_filename: model.pdmodel + params_filename: model.pdiparams + +Distillation: + alpha: 1.0 + loss: soft_label + +Quantization: + use_pact: true + activation_quantize_type: 'moving_average_abs_max' + quantize_op_types: + - conv2d + - depthwise_conv2d + +TrainConfig: + train_iter: 5000 + eval_iter: 1000 + learning_rate: + type: CosineAnnealingDecay + learning_rate: 0.00003 + T_max: 6000 + optimizer_builder: + optimizer: + type: SGD + weight_decay: 4.0e-05 + -- GitLab