diff --git a/tests/ocr_det_params.txt b/tests/ocr_det_params.txt index 54a55839536a05b567d29d763a57d27545187ed1..73b12cec9c4f8a8745f34b00222f89ba68ff9d5f 100644 --- a/tests/ocr_det_params.txt +++ b/tests/ocr_det_params.txt @@ -15,7 +15,7 @@ null:null trainer:norm_train|pact_train norm_train:tools/train.py -c configs/det/det_mv3_db.yml -o Global.pretrained_model=./pretrain_models/MobileNetV3_large_x0_5_pretrained pact_train:deploy/slim/quantization/quant.py -c configs/det/det_mv3_db.yml -o -fpgm_train:deploy/slim/prune/sensitivity_anal.py -c configs/det/det_mv3_db.yml -o +fpgm_train:deploy/slim/prune/sensitivity_anal.py -c configs/det/det_mv3_db.yml -o Global.pretrained_model=./pretrain_models/det_mv3_db_v2.0_train/best_accuracy distill_train:null null:null null:null diff --git a/tests/prepare.sh b/tests/prepare.sh index 0845df5d36bbb2539aa240fcfc4b7b0b6c891ef7..5da74d949f5543f50a4dc61b3093aa11c0caabf6 100644 --- a/tests/prepare.sh +++ b/tests/prepare.sh @@ -34,6 +34,8 @@ MODE=$2 if [ ${MODE} = "lite_train_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 + wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/en/det_mv3_db_v2.0_train.tar + cd ./pretrain_models/ && tar xf det_mv3_db_v2.0_train.tar && cd ../ rm -rf ./train_data/icdar2015 rm -rf ./train_data/ic15_data wget -nc -P ./train_data/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/icdar2015_lite.tar