- dict/kr_lp_dict.txt: 67 chars covering 4 plate types (10 digits + 40 usage hangul + 17 region hangul, dedup) - configs/kr_lp_pgnet.yml: PGNet config tuned for Korean LP (pad_num=67, max_text_length=10, valid_set=partvgg, infer_visual_type=CN) - setup_server.sh: symlink dict and config into PaddleOCR tree Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>main
parent
35c9f9cb5d
commit
3e5f823dd5
@ -0,0 +1,120 @@
|
||||
Global:
|
||||
use_gpu: True
|
||||
epoch_num: 200
|
||||
log_smooth_window: 20
|
||||
print_batch_step: 10
|
||||
save_model_dir: ./output/kr_lp_pgnet/
|
||||
save_epoch_step: 10
|
||||
eval_batch_step: [0, 1000]
|
||||
cal_metric_during_train: False
|
||||
pretrained_model:
|
||||
checkpoints:
|
||||
save_inference_dir:
|
||||
use_visualdl: False
|
||||
infer_img:
|
||||
infer_visual_type: CN
|
||||
valid_set: partvgg
|
||||
save_res_path: ./output/kr_lp_pgnet/predicts.txt
|
||||
character_dict_path: ppocr/utils/kr_lp_dict.txt
|
||||
character_type: CN
|
||||
max_text_length: 10
|
||||
max_text_nums: 5
|
||||
tcl_len: 64
|
||||
|
||||
Architecture:
|
||||
model_type: e2e
|
||||
algorithm: PGNet
|
||||
Transform:
|
||||
Backbone:
|
||||
name: ResNet
|
||||
layers: 50
|
||||
Neck:
|
||||
name: PGFPN
|
||||
Head:
|
||||
name: PGHead
|
||||
character_dict_path: ppocr/utils/kr_lp_dict.txt
|
||||
|
||||
Loss:
|
||||
name: PGLoss
|
||||
tcl_bs: 64
|
||||
max_text_length: 10
|
||||
max_text_nums: 5
|
||||
pad_num: 67
|
||||
|
||||
Optimizer:
|
||||
name: Adam
|
||||
beta1: 0.9
|
||||
beta2: 0.999
|
||||
lr:
|
||||
name: Cosine
|
||||
learning_rate: 0.001
|
||||
warmup_epoch: 50
|
||||
regularizer:
|
||||
name: 'L2'
|
||||
factor: 0.0001
|
||||
|
||||
PostProcess:
|
||||
name: PGPostProcess
|
||||
score_thresh: 0.5
|
||||
mode: fast
|
||||
point_gather_mode: align
|
||||
|
||||
Metric:
|
||||
name: E2EMetric
|
||||
mode: A
|
||||
gt_mat_dir: ./train_data/kr_lp_synth/gt
|
||||
character_dict_path: ppocr/utils/kr_lp_dict.txt
|
||||
main_indicator: f_score_e2e
|
||||
|
||||
Train:
|
||||
dataset:
|
||||
name: PGDataSet
|
||||
data_dir: ./train_data/kr_lp_synth/train
|
||||
label_file_list: [./train_data/kr_lp_synth/train/train.txt]
|
||||
ratio_list: [1.0]
|
||||
transforms:
|
||||
- DecodeImage:
|
||||
img_mode: BGR
|
||||
channel_first: False
|
||||
- E2ELabelEncodeTrain:
|
||||
- PGProcessTrain:
|
||||
batch_size: 14
|
||||
use_resize: True
|
||||
use_random_crop: False
|
||||
min_crop_size: 24
|
||||
min_text_size: 4
|
||||
max_text_size: 512
|
||||
point_gather_mode: align
|
||||
- KeepKeys:
|
||||
keep_keys: ['images', 'tcl_maps', 'tcl_label_maps', 'border_maps', 'direction_maps', 'training_masks', 'label_list', 'pos_list', 'pos_mask']
|
||||
loader:
|
||||
shuffle: True
|
||||
drop_last: True
|
||||
batch_size_per_card: 14
|
||||
num_workers: 8
|
||||
|
||||
Eval:
|
||||
dataset:
|
||||
name: PGDataSet
|
||||
data_dir: ./train_data/kr_lp_synth/test
|
||||
label_file_list: [./train_data/kr_lp_synth/test/test.txt]
|
||||
transforms:
|
||||
- DecodeImage:
|
||||
img_mode: BGR
|
||||
channel_first: False
|
||||
- E2ELabelEncodeTest:
|
||||
- E2EResizeForTest:
|
||||
max_side_len: 768
|
||||
- NormalizeImage:
|
||||
scale: 1./255.
|
||||
mean: [0.485, 0.456, 0.406]
|
||||
std: [0.229, 0.224, 0.225]
|
||||
order: 'hwc'
|
||||
- ToCHWImage:
|
||||
- KeepKeys:
|
||||
keep_keys: ['image', 'shape', 'polys', 'texts', 'ignore_tags', 'img_id']
|
||||
loader:
|
||||
shuffle: False
|
||||
drop_last: False
|
||||
batch_size_per_card: 1
|
||||
num_workers: 2
|
||||
@ -0,0 +1,67 @@
|
||||
0
|
||||
1
|
||||
2
|
||||
3
|
||||
4
|
||||
5
|
||||
6
|
||||
7
|
||||
8
|
||||
9
|
||||
가
|
||||
나
|
||||
다
|
||||
라
|
||||
마
|
||||
바
|
||||
사
|
||||
아
|
||||
자
|
||||
거
|
||||
너
|
||||
더
|
||||
러
|
||||
머
|
||||
버
|
||||
서
|
||||
어
|
||||
저
|
||||
고
|
||||
노
|
||||
도
|
||||
로
|
||||
모
|
||||
보
|
||||
소
|
||||
오
|
||||
조
|
||||
구
|
||||
누
|
||||
두
|
||||
루
|
||||
무
|
||||
부
|
||||
수
|
||||
우
|
||||
주
|
||||
하
|
||||
허
|
||||
호
|
||||
배
|
||||
울
|
||||
산
|
||||
대
|
||||
인
|
||||
천
|
||||
광
|
||||
전
|
||||
세
|
||||
종
|
||||
경
|
||||
기
|
||||
강
|
||||
원
|
||||
충
|
||||
북
|
||||
남
|
||||
제
|
||||
Loading…
Reference in new issue