diff --git a/test_tipc/configs/det_r50_db++/train_infer_python.txt b/test_tipc/configs/det_r50_db_plusplus/train_infer_python.txt similarity index 98% rename from test_tipc/configs/det_r50_db++/train_infer_python.txt rename to test_tipc/configs/det_r50_db_plusplus/train_infer_python.txt index bcf393a52b0e073267aa7423960179d8b5eba4bd..04a3e845859167f78d5b3dd799236f8b8a051e81 100644 --- a/test_tipc/configs/det_r50_db++/train_infer_python.txt +++ b/test_tipc/configs/det_r50_db_plusplus/train_infer_python.txt @@ -1,5 +1,5 @@ ===========================train_params=========================== -model_name:det_r50_db++ +model_name:det_r50_db_plusplus python:python3.7 gpu_list:0|0,1 Global.use_gpu:True|True diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh index eb6d1b2025cf3a008b8db2ab1b31e98abb614c78..73016ca44ae6c22e7a81b71af9efed4ea255f42a 100644 --- a/test_tipc/prepare.sh +++ b/test_tipc/prepare.sh @@ -59,7 +59,7 @@ if [ ${MODE} = "lite_train_lite_infer" ];then wget -nc -P ./inference/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/table/en_ppocr_mobile_v2.0_table_rec_infer.tar --no-check-certificate cd ./inference/ && tar xf en_ppocr_mobile_v2.0_table_det_infer.tar && tar xf en_ppocr_mobile_v2.0_table_rec_infer.tar && cd ../ fi - if [[ ${model_name} =~ "det_r50_db++" ]];then + if [[ ${model_name} =~ "det_r50_db_plusplus" ]];then wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/dygraph_v2.1/en_det/ResNet50_dcn_asf_synthtext_pretrained.pdparams --no-check-certificate fi if [ ${model_name} == "table_master" ];then