diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh index 9813d9529c2372699ba3d49b15506e59b179d442..daf12f429ac81ebfb165ee68c0c42433a05691bc 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 123c6613cde81078aa585c2d2d01d6289e4cf426..21bbee098ef19456d05165969a9ad400400f1264 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],