Merge pull request #1639 from WenmuZhou/dygraph_rc

add tensorrt to hubserving
PaddlePaddle · Jan 4, 2021 · 5df41e8 · 5df41e8
2 parents 67ef25d + c465f40
commit 5df41e8
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/deploy/hubserving/ocr_cls/params.py b/deploy/hubserving/ocr_cls/params.py
@@ -18,7 +18,7 @@ def read_params():
  cfg.cls_batch_num = 30
  cfg.cls_thresh = 0.9
 
- cfg.use_zero_copy_run = False
  cfg.use_pdserving = False
+ cfg.use_tensorrt = False
 
  return cfg
diff --git a/deploy/hubserving/ocr_det/params.py b/deploy/hubserving/ocr_det/params.py
@@ -27,7 +27,7 @@ def read_params():
  # cfg.det_east_cover_thresh = 0.1
  # cfg.det_east_nms_thresh = 0.2
 
- cfg.use_zero_copy_run = False
  cfg.use_pdserving = False
+ cfg.use_tensorrt = False
 
  return cfg
diff --git a/deploy/hubserving/ocr_rec/params.py b/deploy/hubserving/ocr_rec/params.py
@@ -13,7 +13,7 @@ def read_params():
 
  #params for text recognizer
  cfg.rec_algorithm = "CRNN"
- cfg.rec_model_dir = "./inference/ch_ppocr_mobile_v1.1_rec_infer/"
+ cfg.rec_model_dir = "./inference/ch_ppocr_mobile_v2.0_rec_infer/"
 
  cfg.rec_image_shape = "3, 32, 320"
  cfg.rec_char_type = 'ch'
@@ -23,7 +23,7 @@ def read_params():
  cfg.rec_char_dict_path = "./ppocr/utils/ppocr_keys_v1.txt"
  cfg.use_space_char = True
 
- cfg.use_zero_copy_run = False
  cfg.use_pdserving = False
+ cfg.use_tensorrt = False
 
  return cfg
diff --git a/deploy/hubserving/ocr_system/params.py b/deploy/hubserving/ocr_system/params.py
@@ -47,8 +47,8 @@ def read_params():
  cfg.cls_batch_num = 30
  cfg.cls_thresh = 0.9
 
- cfg.use_zero_copy_run = False
  cfg.use_pdserving = False
+ cfg.use_tensorrt = False
  cfg.drop_score = 0.5
 
  return cfg