From 418385b88c6f83a3f2b91aded7f5f4e952e8e0c8 Mon Sep 17 00:00:00 2001 From: tink2123 Date: Tue, 17 Aug 2021 07:20:30 +0000 Subject: [PATCH] smaller rec_inference dataset --- tests/ocr_rec_params.txt | 2 +- tests/prepare.sh | 7 ++++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/ocr_rec_params.txt b/tests/ocr_rec_params.txt index cc558844..123719f9 100644 --- a/tests/ocr_rec_params.txt +++ b/tests/ocr_rec_params.txt @@ -9,7 +9,7 @@ Global.save_model_dir:./output/ Train.loader.batch_size_per_card:lite_train_infer=128|whole_train_infer=128 Global.pretrained_model:null train_model_name:latest -train_infer_img_dir:./train_data/ic15_data/test +train_infer_img_dir:./inference/rec_inference null:null ## trainer:norm_train|pact_train diff --git a/tests/prepare.sh b/tests/prepare.sh index 8f5291da..828705c3 100644 --- a/tests/prepare.sh +++ b/tests/prepare.sh @@ -38,10 +38,11 @@ if [ ${MODE} = "lite_train_infer" ];then rm -rf ./train_data/ic15_data wget -nc -P ./train_data/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/icdar2015_lite.tar wget -nc -P ./train_data/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/ic15_data.tar # todo change to bcebos - + wget -nc -P ./inference https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/rec_inference.tar cd ./train_data/ && tar xf icdar2015_lite.tar && tar xf ic15_data.tar ln -s ./icdar2015_lite ./icdar2015 cd ../ + cd ./inference && tar xf rec_inference.tar && cd ../ elif [ ${MODE} = "whole_train_infer" ];then wget -nc -P ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x0_5_pretrained.pdparams rm -rf ./train_data/icdar2015 @@ -68,9 +69,9 @@ else else rm -rf ./train_data/ic15_data eval_model_name="ch_ppocr_mobile_v2.0_rec_infer" - wget -nc -P ./train_data/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/ic15_data.tar + wget -nc -P ./inference/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/rec_inference.tar wget -nc -P ./inference https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_rec_infer.tar - cd ./inference && tar xf ${eval_model_name}.tar && tar xf ic15_data.tar && cd ../ + cd ./inference && tar xf ${eval_model_name}.tar && tar xf rec_inference.tar && cd ../ fi fi -- GitLab