From 8a130c4545df2ecc6efccf039128cd5cc8bf5269 Mon Sep 17 00:00:00 2001
From: andyjpaddle <jiangkaitao@baidu.com>
Date: Fri, 10 Dec 2021 08:24:09 +0000
Subject: [PATCH] fix some wrong dir name

---
 test_tipc/configs/ch_PP-OCRv2_rec/train_infer_python.txt     | 4 ++--
 .../ch_ppocr_mobile_v2.0_rec_FPGM/train_infer_python.txt     | 2 +-
 .../ch_ppocr_mobile_v2.0_rec_PACT/train_infer_python.txt     | 2 +-
 test_tipc/prepare.sh                                         | 5 +++++
 4 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/test_tipc/configs/ch_PP-OCRv2_rec/train_infer_python.txt b/test_tipc/configs/ch_PP-OCRv2_rec/train_infer_python.txt
index 375bd13a..4607b0a7 100644
--- a/test_tipc/configs/ch_PP-OCRv2_rec/train_infer_python.txt
+++ b/test_tipc/configs/ch_PP-OCRv2_rec/train_infer_python.txt
@@ -34,7 +34,7 @@ distill_export:null
 export1:null
 export2:null
 inference_dir:Student
-infer_model:./inference/ch_PP-OCRv2_rec_infer/
+infer_model:./inference/ch_PP-OCRv2_rec_infer
 infer_export:null
 infer_quant:False
 inference:tools/infer/predict_rec.py
@@ -45,7 +45,7 @@ inference:tools/infer/predict_rec.py
 --use_tensorrt:False|True
 --precision:fp32|fp16|int8
 --rec_model_dir:
---image_dir:/inference/rec_inference
+--image_dir:./inference/rec_inference
 null:null
 --benchmark:True
 null:null
diff --git a/test_tipc/configs/ch_ppocr_mobile_v2.0_rec_FPGM/train_infer_python.txt b/test_tipc/configs/ch_ppocr_mobile_v2.0_rec_FPGM/train_infer_python.txt
index 90dd71c4..a968573d 100644
--- a/test_tipc/configs/ch_ppocr_mobile_v2.0_rec_FPGM/train_infer_python.txt
+++ b/test_tipc/configs/ch_ppocr_mobile_v2.0_rec_FPGM/train_infer_python.txt
@@ -34,7 +34,7 @@ distill_export:null
 export1:null
 export2:null
 inference_dir:null
-train_model:./inference/ch_PP-OCRv2_rec_train/best_accuracy
+train_model:null
 infer_export:null
 infer_quant:False
 inference:tools/infer/predict_rec.py
diff --git a/test_tipc/configs/ch_ppocr_mobile_v2.0_rec_PACT/train_infer_python.txt b/test_tipc/configs/ch_ppocr_mobile_v2.0_rec_PACT/train_infer_python.txt
index 0890b35d..56b9e189 100644
--- a/test_tipc/configs/ch_ppocr_mobile_v2.0_rec_PACT/train_infer_python.txt
+++ b/test_tipc/configs/ch_ppocr_mobile_v2.0_rec_PACT/train_infer_python.txt
@@ -34,7 +34,7 @@ distill_export:null
 export1:null
 export2:null
 inference_dir:null
-train_model:./inference/ch_PP-OCRv2_rec_slim_quant_train/best_accuracy
+train_model:null
 infer_export:null
 infer_quant:False
 inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ppocr_keys_v1.txt --rec_image_shape="3,32,100"
diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
index dae2c1ca..6b5780c8 100644
--- a/test_tipc/prepare.sh
+++ b/test_tipc/prepare.sh
@@ -143,6 +143,11 @@ elif [ ${MODE} = "whole_infer" ];then
         eval_model_name="ch_PP-OCRv2_det_infer"
         wget -nc -P ./inference/ https://paddleocr.bj.bcebos.com/PP-OCRv2/chinese/ch_PP-OCRv2_det_infer.tar --no-check-certificate
         cd ./inference && tar xf ${eval_model_name}.tar && tar xf ch_det_data_50.tar && cd ../
+    fi
+    if [ ${model_name} == "PPOCRv2_ocr_rec" ]; then
+        eval_model_name="ch_PP-OCRv2_rec_infer"
+        wget -nc -P ./inference/ https://paddleocr.bj.bcebos.com/PP-OCRv2/chinese/ch_PP-OCRv2_rec_infer.tar --no-check-certificate
+        cd ./inference && tar xf ${eval_model_name}.tar && cd ../
     fi   
     if [ ${model_name} == "en_server_pgnetA" ]; then
         wget -nc -P ./inference/  https://paddleocr.bj.bcebos.com/dygraph_v2.0/pgnet/en_server_pgnetA.tar  --no-check-certificate
-- 
GitLab