diff --git a/test_tipc/configs/sr_telescope/sr_telescope.yml b/test_tipc/configs/sr_telescope/sr_telescope.yml index d3c10448e423ff0305950ea39664379e60f8a113..c78a42d0efb7bbcdd182861a6474d87c9f68b3d4 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 4dcfa29ee146b3b2662122966d859142bb0ed0c5..7235f07e8c72411f6ae979e666e624c32de935b9 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 452ec31c48f2b5bd807a3a0a7c6bd0d95b520e28..02ee8a24d241195d1330ea42fc05ed35dd7a87b7 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