diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh index 701a91c53ac69652cb80b5af3067e6be9c514fb6..910ab5c5e0083226a33e5a353e17b9780e11ea6f 100644 --- a/test_tipc/prepare.sh +++ b/test_tipc/prepare.sh @@ -20,6 +20,14 @@ model_name=$(func_parser_value "${lines[1]}") trainer_list=$(func_parser_value "${lines[14]}") +if [ ${MODE} = "benchmark_train" ];then + if [[ ${model_name} =~ "det_mv3_db_v2.0_benchmark" ]];then + rm -rf ./train_data/icdar2015 + 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 ./train_data/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/icdar2015.tar --no-check-certificate + cd ./train_data/ && tar xf icdar2015.tar && cd ../ + fi +fi if [ ${MODE} = "lite_train_lite_infer" ];then # pretrain lite train data @@ -65,11 +73,6 @@ if [ ${MODE} = "lite_train_lite_infer" ];then wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_rec_train.tar --no-check-certificate cd ./pretrain_models/ && tar xf ch_ppocr_mobile_v2.0_rec_train.tar && cd ../ fi - if [[ ${model_name} =~ "det_mv3_db_v2.0_benchmark" ]];then - rm -rf ./train_data/icdar2015 - wget -nc -P ./train_data/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/icdar2015.tar --no-check-certificate - cd ./train_data/ && tar xf icdar2015.tar && cd ../ - fi elif [ ${MODE} = "whole_train_whole_infer" ];then wget -nc -P ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x0_5_pretrained.pdparams --no-check-certificate