From b0feca82d549b8f1a4d32d0cb2021f0df654fff5 Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
Date: Thu, 9 Sep 2021 08:28:58 +0000
Subject: [PATCH] fix nonfinite

---
 tests/configs/det_mv3_db.yml | 6 +++---
 tests/ocr_det_params.txt     | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/configs/det_mv3_db.yml b/tests/configs/det_mv3_db.yml
index d6d4c26c..5eada6d5 100644
--- a/tests/configs/det_mv3_db.yml
+++ b/tests/configs/det_mv3_db.yml
@@ -23,10 +23,10 @@ Architecture:
     name: MobileNetV3
     scale: 0.5
     model_name: large
-    disable_se: True
+    disable_se: False
   Neck:
     name: DBFPN
-    out_channels: 96
+    out_channels: 256
   Head:
     name: DBHead
     k: 50
@@ -74,7 +74,7 @@ Train:
           channel_first: False
       - DetLabelEncode: # Class handling label
       - Resize:
-          # size: [640, 640]
+          size: [640, 640]
       - MakeBorderMap:
           shrink_ratio: 0.4
           thresh_min: 0.3
diff --git a/tests/ocr_det_params.txt b/tests/ocr_det_params.txt
index 7be1430a..bbe6a604 100644
--- a/tests/ocr_det_params.txt
+++ b/tests/ocr_det_params.txt
@@ -35,7 +35,7 @@ export1:null
 export2:null
 ##
 train_model:./inference/ch_ppocr_mobile_v2.0_det_train/best_accuracy
-infer_export:tools/export_model.py -c configs/det/det_mv3_db.yml -o
+infer_export:tools/export_model.py -c tests/configs/ch_ppocr_v2.0/ch_det_mv3_db_v2.0.yml -o
 infer_quant:False
 inference:tools/infer/predict_det.py
 --use_gpu:True|False
-- 
GitLab