提交 67c0a67b 编写于 作者: L LDOUBLEV

fix

上级 b5ed094e
......@@ -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
......
......@@ -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],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册