"vscode:/vscode.git/clone" did not exist on "539b416c4801a540b5c8c09b16fdd6d3d47bba3c"
Commit 048f6af4 authored by myhloli's avatar myhloli
Browse files

feat(ocr): add PP-OCRv5 models and update configurations

- Add new PP-OCRv5 detection and recognition models
- Update arch_config.yaml with new model architectures
- Modify models_config.yml to include PP-OCRv5 models for ch_lite configuration- Change dictionary file for ch_lite to ppocrv5_dict.txt
parent 002333a8
...@@ -104,6 +104,22 @@ ch_PP-OCRv4_det_infer: ...@@ -104,6 +104,22 @@ ch_PP-OCRv4_det_infer:
name: DBHead name: DBHead
k: 50 k: 50
ch_PP-OCRv5_det_infer:
model_type: det
algorithm: DB
Transform: null
Backbone:
name: PPLCNetV3
scale: 0.75
det: True
Neck:
name: RSEFPN
out_channels: 96
shortcut: True
Head:
name: DBHead
k: 50
ch_PP-OCRv4_det_server_infer: ch_PP-OCRv4_det_server_infer:
model_type: det model_type: det
algorithm: DB algorithm: DB
...@@ -196,6 +212,32 @@ ch_PP-OCRv4_rec_server_doc_infer: ...@@ -196,6 +212,32 @@ ch_PP-OCRv4_rec_server_doc_infer:
nrtr_dim: 384 nrtr_dim: 384
max_text_length: 25 max_text_length: 25
ch_PP-OCRv5_rec_infer:
model_type: rec
algorithm: SVTR_HGNet
Transform:
Backbone:
name: PPLCNetV3
scale: 0.95
Head:
name: MultiHead
out_channels_list:
CTCLabelDecode: 18385
head_list:
- CTCHead:
Neck:
name: svtr
dims: 120
depth: 2
hidden_dims: 120
kernel_size: [ 1, 3 ]
use_guide: True
Head:
fc_decay: 0.00001
- NRTRHead:
nrtr_dim: 384
max_text_length: 25
chinese_cht_PP-OCRv3_rec_infer: chinese_cht_PP-OCRv3_rec_infer:
model_type: rec model_type: rec
algorithm: SVTR algorithm: SVTR
......
lang: lang:
ch_lite: ch_lite_v4:
det: ch_PP-OCRv3_det_infer.pth det: ch_PP-OCRv3_det_infer.pth
rec: ch_PP-OCRv4_rec_infer.pth rec: ch_PP-OCRv4_rec_infer.pth
dict: ppocr_keys_v1.txt dict: ppocr_keys_v1.txt
ch_lite:
det: ch_PP-OCRv5_det_infer.pth
rec: ch_PP-OCRv5_rec_infer.pth
dict: ppocrv5_dict.txt
ch_server: ch_server:
det: ch_PP-OCRv3_det_infer.pth det: ch_PP-OCRv3_det_infer.pth
rec: ch_PP-OCRv4_rec_server_infer.pth rec: ch_PP-OCRv4_rec_server_infer.pth
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment