From fb33880e8e52fa953e97b32541751c534f547982 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Mon, 27 Dec 2021 15:38:19 +0800 Subject: [PATCH] add distill det doc --- configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_cml.yml | 3 +++ configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_distill.yml | 1 + configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_dml.yml | 1 + 3 files changed, 5 insertions(+) diff --git a/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_cml.yml b/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_cml.yml index bb6a19686..b75ca6fdc 100644 --- a/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_cml.yml +++ b/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_cml.yml @@ -21,6 +21,7 @@ Architecture: model_type: det Models: Teacher: + pretrained: ./pretrain_models/ch_ppocr_server_v2.0_det_train/best_accuracy freeze_params: true return_all_feats: false model_type: det @@ -36,6 +37,7 @@ Architecture: name: DBHead k: 50 Student: + pretrained: freeze_params: false return_all_feats: false model_type: det @@ -52,6 +54,7 @@ Architecture: name: DBHead k: 50 Student2: + pretrained: freeze_params: false return_all_feats: false model_type: det diff --git a/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_distill.yml b/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_distill.yml index d8d5135dd..b4644244f 100644 --- a/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_distill.yml +++ b/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_distill.yml @@ -18,6 +18,7 @@ Global: Architecture: name: DistillationModel algorithm: Distillation + model_type: det Models: Student: pretrained: ./pretrain_models/MobileNetV3_large_x0_5_pretrained diff --git a/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_dml.yml b/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_dml.yml index bfbc3b626..0553a65e4 100644 --- a/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_dml.yml +++ b/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_dml.yml @@ -18,6 +18,7 @@ Global: Architecture: name: DistillationModel algorithm: Distillation + model_type: det Models: Student: pretrained: ./pretrain_models/MobileNetV3_large_x0_5_pretrained