From 85a64cb0cb80f50c76510c95a0d87766c0afffbb Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Fri, 25 Mar 2022 14:54:44 +0800 Subject: [PATCH] fix tipc --- configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_cml.yml | 2 +- test_tipc/prepare.sh | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) 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 b75ca6fd..3833cb0b 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 c950c059..dd66b3db 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 -- GitLab