47 lines
1.1 KiB
YAML
47 lines
1.1 KiB
YAML
# params for prediction engine
|
|
use_gpu: true
|
|
ir_optim: true
|
|
enable_mkldnn: false
|
|
# use_zero_copy_run: true
|
|
use_tensorrt: false
|
|
num_cpu_threads: 6
|
|
gpu_id: 0
|
|
gpu_mem: 2000
|
|
|
|
# params for text detector
|
|
det_algorithm: "DB"
|
|
det_model_dir: "https://paddleocr.bj.bcebos.com/20-09-22/mobile/det/ch_ppocr_mobile_v1.1_det_infer.tar"
|
|
det_max_side_len: 960
|
|
|
|
# DB parmas
|
|
det_db_thresh: 0.3
|
|
det_db_box_thresh: 0.5
|
|
det_db_unclip_ratio: 2.0
|
|
|
|
# EAST parmas
|
|
det_east_score_thresh: 0.8
|
|
det_east_cover_thresh: 0.1
|
|
det_east_nms_thresh: 0.2
|
|
|
|
# params for text recognizer
|
|
rec_algorithm: "CRNN"
|
|
rec_model_dir: "https://paddleocr.bj.bcebos.com/20-09-22/mobile/rec/ch_ppocr_mobile_v1.1_rec_infer.tar"
|
|
rec_image_shape: [3, 32, 320]
|
|
rec_char_type: "ch"
|
|
rec_batch_num: 30
|
|
max_text_length: 25
|
|
rec_char_dict_path: "config/ppocr_keys_v1.txt"
|
|
use_space_char: true
|
|
|
|
# params for text classifier
|
|
use_angle_cls: false
|
|
cls_model_dir: "https://paddleocr.bj.bcebos.com/20-09-22/cls/ch_ppocr_mobile_v1.1_cls_infer.tar"
|
|
cls_image_shape: [3, 48, 192]
|
|
label_list: ["0", "180"]
|
|
cls_batch_num: 30
|
|
cls_thresh: 0.9
|
|
|
|
lang: ch
|
|
det: true
|
|
rec: true
|
|
cls: false |