diff --git a/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_cml.yml b/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_cml.yml index b75ca6fdc7e7543460ba2e561af27e4d4fbf3391..3833cb0bad9915a2169b116e7406e01cadd0ef62 100644 --- a/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_cml.yml +++ b/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_cml.yml @@ -8,7 +8,7 @@ Global: # evaluation is run every 5000 iterations after the 4000th iteration eval_batch_step: [3000, 2000] cal_metric_during_train: False - pretrained_model: ./pretrain_models/ch_PP-OCRv2_det_distill_train/best_accuracy + pretrained_model: checkpoints: save_inference_dir: use_visualdl: False diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh index c950c059e7154d2b856f0ee7a7d6e0a5de67b938..dd66b3dbf94391e472c96d409291398245337d93 100644 --- a/test_tipc/prepare.sh +++ b/test_tipc/prepare.sh @@ -60,7 +60,11 @@ if [ ${MODE} = "lite_train_lite_infer" ];then ln -s ./icdar2015_lite ./icdar2015 cd ../ cd ./inference && tar xf rec_inference.tar && cd ../ - if [ ${model_name} == "" ]; then + if [ ${model_name} == "ch_PPOCRv2_det" ]; then + wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_det_train.tar --no-check-certificate + cd ./pretrain_models/ && tar xf ch_ppocr_mobile_v2.0_det_train.tar && cd ../ + fi + if [ ${model_name} == "det_r18_db_v2_0" ]; then wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/pretrained/ResNet18_vd_pretrained.pdparams --no-check-certificate fi if [ ${model_name} == "en_server_pgnetA" ]; then