From af4aa1aa2ac4aefee7417fb15e92815994eb0b2a Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
Date: Wed, 27 Apr 2022 19:06:24 +0800
Subject: [PATCH] fix postprocess

---
 configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml b/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml
index 3efe314f..6c60e12f 100644
--- a/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml
+++ b/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml
@@ -116,15 +116,22 @@ Optimizer:
   regularizer:
     name: L2
     factor: 5.0e-05
+
 PostProcess:
-  name: DBPostProcess
+  name: DistillationDBPostProcess
+  model_name: ["Student"]
+  key: head_out
   thresh: 0.3
   box_thresh: 0.6
   max_candidates: 1000
   unclip_ratio: 1.5
+
 Metric:
-  name: DetMetric
+  name: DistillationMetric
+  base_metric_name: DetMetric
   main_indicator: hmean
+  key: "Student"
+
 Train:
   dataset:
     name: SimpleDataSet
-- 
GitLab