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: Global:
text_score: 0.5 text_score: 0.5
use_det: true use_det: true
use_cls: true use_cls: true
use_rec: true use_rec: true
print_verbose: false print_verbose: false
min_height: 30 min_height: 30
width_height_ratio: 40 width_height_ratio: 40
max_side_len: 1500
min_side_len: 30
Det: 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_side_len: 1500
limit_type: min limit_type: min
thresh: 0.3 thresh: 0.3
box_thresh: 0.3 box_thresh: 0.3
max_candidates: 1000 max_candidates: 1000
unclip_ratio: 1.6 unclip_ratio: 1.6
use_dilation: true use_dilation: true
score_mode: fast score_mode: fast
Cls: 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_image_shape: [3, 48, 192]
cls_batch_num: 6 cls_batch_num: 6
cls_thresh: 0.9 cls_thresh: 0.9
label_list: ['0', '180'] label_list: ["0", "180"]
Rec: 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_img_shape: [3, 48, 320]
rec_batch_num: 6 rec_batch_num: 6

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

@ -7,6 +7,8 @@ Global:
min_height: 30 min_height: 30
width_height_ratio: 40 width_height_ratio: 40
use_space_char: true use_space_char: true
max_side_len: 1500
min_side_len: 30
Det: Det:
use_cuda: false use_cuda: false
@ -39,4 +41,4 @@ Rec:
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_img_shape: [3, 48, 320]
rec_batch_num: 6 rec_batch_num: 6