Merge pull request #7962 from WenmuZhou/tipc3

add pse tipc
2026-09-24 23:33:08 +08:00 · 2022-10-17 19:36:05 +08:00
parent caf77ddb2f c44f3bc78c
commit 28fcdd6033
1 changed files with 3 additions and 0 deletions
@@ -241,6 +241,9 @@ if [ ${MODE} = "lite_train_lite_infer" ];then
    if [ ${model_name} == "ch_ppocr_mobile_v2_0_det_FPGM" ]; then
        ${python_name} -m pip install paddleslim
    fi
+    if [ ${model_name} == "det_r50_vd_pse_v2_0" ]; then
+        wget -nc -P ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/ResNet50_vd_ssld_pretrained.pdparams --no-check-certificate
+    fi
    if [ ${model_name} == "det_mv3_east_v2_0" ]; then
        wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/en/det_mv3_east_v2.0_train.tar --no-check-certificate
        cd ./pretrain_models/ && tar xf det_mv3_east_v2.0_train.tar && cd ../