fix
parent
b5ed094e20
commit
67c0a67bec
|
@ -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],
|
||||
|
|
Loading…
Reference in New Issue