payoff_OCR/ocr.yaml

45 lines
1.1 KiB
YAML

pipeline_name: OCR
text_type: general
use_doc_preprocessor: False
use_textline_orientation: False
SubPipelines:
DocPreprocessor:
pipeline_name: doc_preprocessor
use_doc_orientation_classify: False
use_doc_unwarping: True
SubModules:
DocOrientationClassify:
module_name: doc_text_orientation
model_name: PP-LCNet_x1_0_doc_ori
model_dir: null
DocUnwarping:
module_name: image_unwarping
model_name: UVDoc
model_dir: null
SubModules:
TextDetection:
module_name: text_detection
model_name: PP-OCRv4_mobile_det
model_dir: ./model/PP-OCRv4_mobile_det
limit_side_len: 960
limit_type: max
thresh: 0.3 # 0.3
box_thresh: 0.6 # 0.6
unclip_ratio: 2.0
TextLineOrientation:
module_name: textline_orientation
model_name: PP-LCNet_x0_25_textline_ori
model_dir: null
batch_size: 6
TextRecognition:
module_name: text_recognition
model_name: PP-OCRv4_mobile_rec
model_dir: ./model/PP-OCRv4_mobile_rec
batch_size: 6
score_thresh: 0.0