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