From d1f57063fb419bc75d0889d20010786c8c9e7083 Mon Sep 17 00:00:00 2001
From: WenmuZhou <572459439@qq.com>
Date: Tue, 25 Oct 2022 06:30:25 +0000
Subject: [PATCH] fix telescope tipc error

---
 test_tipc/configs/sr_telescope/sr_telescope.yml       |  2 +-
 test_tipc/configs/sr_telescope/train_infer_python.txt | 10 +++++-----
 test_tipc/prepare.sh                                  |  2 +-
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/test_tipc/configs/sr_telescope/sr_telescope.yml b/test_tipc/configs/sr_telescope/sr_telescope.yml
index d3c10448..c78a42d0 100644
--- a/test_tipc/configs/sr_telescope/sr_telescope.yml
+++ b/test_tipc/configs/sr_telescope/sr_telescope.yml
@@ -51,7 +51,7 @@ Metric:
 Train:
   dataset:
     name: LMDBDataSetSR
-    data_dir: ./train_data/TextZoom/train
+    data_dir: ./train_data/TextZoom/test
     transforms:
       - SRResize:
           imgH: 32
diff --git a/test_tipc/configs/sr_telescope/train_infer_python.txt b/test_tipc/configs/sr_telescope/train_infer_python.txt
index 4dcfa29e..7235f07e 100644
--- a/test_tipc/configs/sr_telescope/train_infer_python.txt
+++ b/test_tipc/configs/sr_telescope/train_infer_python.txt
@@ -4,12 +4,12 @@ python:python3.7
 gpu_list:0|0,1
 Global.use_gpu:True|True
 Global.auto_cast:null
-Global.epoch_num:lite_train_lite_infer=2|whole_train_whole_infer=300
+Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=300
 Global.save_model_dir:./output/
 Train.loader.batch_size_per_card:lite_train_lite_infer=16|whole_train_whole_infer=16
 Global.pretrained_model:null
 train_model_name:latest
-train_infer_img_dir:./inference/sr_inference
+train_infer_img_dir:./inference/rec_inference
 null:null
 ##
 trainer:norm_train
@@ -21,7 +21,7 @@ null:null
 null:null
 ##
 ===========================eval_params===========================
-eval:tools/eval.py -c test_tipc/configs/sr_telescope/sr_telescope.yml -o
+eval:null
 null:null
 ##
 ===========================infer_params===========================
@@ -44,8 +44,8 @@ inference:tools/infer/predict_sr.py --sr_image_shape="1,32,128" --rec_algorithm=
 --rec_batch_num:1
 --use_tensorrt:False
 --precision:fp32
---rec_model_dir:
---image_dir:./inference/sr_inference
+--sr_model_dir:
+--image_dir:./inference/rec_inference
 --save_log_path:./test/output/
 --benchmark:True
 null:null
diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
index 452ec31c..02ee8a24 100644
--- a/test_tipc/prepare.sh
+++ b/test_tipc/prepare.sh
@@ -150,6 +150,7 @@ if [ ${MODE} = "lite_train_lite_infer" ];then
     # pretrain lite train data
     wget -nc -P  ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x0_5_pretrained.pdparams  --no-check-certificate
     wget -nc -P ./pretrain_models/  https://paddleocr.bj.bcebos.com/dygraph_v2.0/en/det_mv3_db_v2.0_train.tar  --no-check-certificate
+    cd ./pretrain_models/ && tar xf det_mv3_db_v2.0_train.tar && cd ../
     if [[ ${model_name} =~ "ch_PP-OCRv2_det" ]];then
         wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/PP-OCRv2/chinese/ch_PP-OCRv2_det_distill_train.tar --no-check-certificate
         cd ./pretrain_models/ && tar xf ch_PP-OCRv2_det_distill_train.tar && cd ../
@@ -179,7 +180,6 @@ if [ ${MODE} = "lite_train_lite_infer" ];then
         wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/ppstructure/models/tablemaster/table_structure_tablemaster_train.tar --no-check-certificate
         cd ./pretrain_models/ && tar xf table_structure_tablemaster_train.tar  && cd ../
     fi
-    cd ./pretrain_models/ && tar xf det_mv3_db_v2.0_train.tar && cd ../
     rm -rf ./train_data/icdar2015
     rm -rf ./train_data/ic15_data
     rm -rf ./train_data/pubtabnet
-- 
GitLab