pull/4856/head
LDOUBLEV 2021-12-07 10:29:53 +00:00
parent b5ed094e20
commit 67c0a67bec
2 changed files with 2 additions and 2 deletions

View File

@ -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

View File

@ -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],