tcsenpai/pensieve
1
0
Fork 0
mirror of https://github.com/tcsenpai/pensieve.git synced 2025-06-06 03:05:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

added max_side_len to the ocr options

Browse Source
This commit is contained in:
tcsenpai 2024-11-17 19:29:04 +01:00
parent 9a0cb78ed4
commit 2ca70ab541
2 changed files with 32 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
memos/plugins/ocr/ppocr-gpu.yaml
View File

@ -1,41 +1,43 @@
Global:
text_score: 0.5
use_det: true
use_cls: true
use_rec: true
print_verbose: false
min_height: 30
width_height_ratio: 40
text_score: 0.5
use_det: true
use_cls: true
use_rec: true
print_verbose: false
min_height: 30
width_height_ratio: 40
max_side_len: 1500
min_side_len: 30
Det:
use_cuda: true
use_cuda: true
model_path: models/ch_PP-OCRv4_det_infer.onnx
model_path: models/ch_PP-OCRv4_det_infer.onnx
limit_side_len: 1500
limit_type: min
limit_side_len: 1500
limit_type: min
thresh: 0.3
box_thresh: 0.3
max_candidates: 1000
unclip_ratio: 1.6
use_dilation: true
score_mode: fast
thresh: 0.3
box_thresh: 0.3
max_candidates: 1000
unclip_ratio: 1.6
use_dilation: true
score_mode: fast
Cls:
use_cuda: true
use_cuda: true
model_path: models/ch_ppocr_mobile_v2.0_cls_train.onnx
model_path: models/ch_ppocr_mobile_v2.0_cls_train.onnx
cls_image_shape: [3, 48, 192]
cls_batch_num: 6
cls_thresh: 0.9
label_list: ['0', '180']
cls_image_shape: [3, 48, 192]
cls_batch_num: 6
cls_thresh: 0.9
label_list: ["0", "180"]
Rec:
use_cuda: true
use_cuda: true
model_path: models/ch_PP-OCRv4_rec_infer.onnx
model_path: models/ch_PP-OCRv4_rec_infer.onnx
rec_img_shape: [3, 48, 320]
rec_batch_num: 6
rec_img_shape: [3, 48, 320]
rec_batch_num: 6

2
memos/plugins/ocr/ppocr.yaml
View File

@ -7,6 +7,8 @@ Global:
min_height: 30
width_height_ratio: 40
use_space_char: true
max_side_len: 1500
min_side_len: 30
Det:
use_cuda: false