From 67c0a67bec9fa5a9ffabe7ed5a1ee1212a68d0e8 Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
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