From af3273759b9b99b416e6767f4d1be904b3b7688a Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Thu, 24 Mar 2022 17:16:01 +0800 Subject: [PATCH 1/2] fix pse postprocess for tipc --- test_tipc/configs/det_mv3_pse_v2.0/det_mv3_pse.yml | 4 ++-- test_tipc/configs/det_r50_vd_pse_v2_0/det_r50_vd_pse.yml | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/test_tipc/configs/det_mv3_pse_v2.0/det_mv3_pse.yml b/test_tipc/configs/det_mv3_pse_v2.0/det_mv3_pse.yml index d37fdcfbb5..27ec3ebeaf 100644 --- a/test_tipc/configs/det_mv3_pse_v2.0/det_mv3_pse.yml +++ b/test_tipc/configs/det_mv3_pse_v2.0/det_mv3_pse.yml @@ -56,7 +56,7 @@ PostProcess: thresh: 0 box_thresh: 0.85 min_area: 16 - box_type: box # 'box' or 'poly' + box_type: quad # 'quad' or 'poly' scale: 1 Metric: @@ -132,4 +132,4 @@ Eval: shuffle: False drop_last: False batch_size_per_card: 1 # must be 1 - num_workers: 8 \ No newline at end of file + num_workers: 8 diff --git a/test_tipc/configs/det_r50_vd_pse_v2_0/det_r50_vd_pse.yml b/test_tipc/configs/det_r50_vd_pse_v2_0/det_r50_vd_pse.yml index 5ebc425271..f7e60fd196 100644 --- a/test_tipc/configs/det_r50_vd_pse_v2_0/det_r50_vd_pse.yml +++ b/test_tipc/configs/det_r50_vd_pse_v2_0/det_r50_vd_pse.yml @@ -55,7 +55,7 @@ PostProcess: thresh: 0 box_thresh: 0.85 min_area: 16 - box_type: box # 'box' or 'poly' + box_type: quad # 'quad' or 'poly' scale: 1 Metric: @@ -131,4 +131,4 @@ Eval: shuffle: False drop_last: False batch_size_per_card: 1 # must be 1 - num_workers: 8 \ No newline at end of file + num_workers: 8 From 00a8735073951a69fefee663e426d0d8946ed9b7 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Thu, 24 Mar 2022 17:52:36 +0800 Subject: [PATCH 2/2] fix prepare.sh --- .../train_infer_python.txt | 0 test_tipc/prepare.sh | 3 +++ 2 files changed, 3 insertions(+) rename test_tipc/configs/{det_r18_vd_v2_0 => det_r18_vd_db_v2_0}/train_infer_python.txt (100%) diff --git a/test_tipc/configs/det_r18_vd_v2_0/train_infer_python.txt b/test_tipc/configs/det_r18_vd_db_v2_0/train_infer_python.txt similarity index 100% rename from test_tipc/configs/det_r18_vd_v2_0/train_infer_python.txt rename to test_tipc/configs/det_r18_vd_db_v2_0/train_infer_python.txt diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh index bd4af1923c..c950c059e7 100644 --- a/test_tipc/prepare.sh +++ b/test_tipc/prepare.sh @@ -60,6 +60,9 @@ 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 + 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 wget -nc -P ./train_data/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/total_text_lite.tar --no-check-certificate wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/pgnet/en_server_pgnetA.tar --no-check-certificate