diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh index 73016ca44ae6c22e7a81b71af9efed4ea255f42a..9c62dc5e2c92161d1e2ee6202d3f121dc5866d76 100644 --- a/test_tipc/prepare.sh +++ b/test_tipc/prepare.sh @@ -28,6 +28,10 @@ if [ ${MODE} = "benchmark_train" ];then 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 + if [[ ${model_name} =~ "det_r50_vd_east_v2_0"; then + wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/en/det_r50_vd_east_v2.0_train.tar --no-check-certificate + cd ./pretrain_models/ && tar xf det_r50_vd_east_v2.0_train.tar && cd ../ + fi if [[ ${model_name} =~ "det_r50_vd_east_v2_0" || ${model_name} =~ "det_r50_vd_pse_v2_0" ]];then wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/pretrained/ResNet50_vd_ssld_pretrained.pdparams --no-check-certificate wget -nc -P ./train_data/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/icdar2015.tar --no-check-certificate