From 67c0a67bec9fa5a9ffabe7ed5a1ee1212a68d0e8 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Tue, 7 Dec 2021 10:29:53 +0000 Subject: [PATCH] fix --- test_tipc/prepare.sh | 2 +- tools/infer/utility.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh index 9813d952..daf12f42 100644 --- a/test_tipc/prepare.sh +++ b/test_tipc/prepare.sh @@ -135,7 +135,7 @@ elif [ ${MODE} = "whole_infer" ];then wget -nc -P ./inference https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_server_v2.0_rec_infer.tar --no-check-certificate cd ./inference && tar xf ${eval_model_name}.tar && tar xf rec_inference.tar && cd ../ fi - if [ ${model_name} = "ch_PPOCRv2_det" ]; then + if [ ${model_name} =~ "ch_PPOCRv2_det" ]; then eval_model_name="ch_PP-OCRv2_det_infer" wget -nc -P ./inference https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/ch_det_data_50.tar --no-check-certificate wget -nc -P ./inference/ https://paddleocr.bj.bcebos.com/PP-OCRv2/chinese/ch_PP-OCRv2_det_infer.tar --no-check-certificate diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 123c6613..21bbee09 100644 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -196,7 +196,7 @@ def create_predictor(args, mode, logger): min_subgraph_size=args.min_subgraph_size) # skip the minmum trt subgraph use_dynamic_shape = True - if mode == "det" and args.rec_algorithm == "DB": + if mode == "det": min_input_shape = { "x": [1, 3, 50, 50], "conv2d_92.tmp_0": [1, 120, 20, 20], -- GitLab