Upload model weights for training
Browse files- .gitattributes +2 -0
- best_accuracy.pdopt +3 -0
- best_accuracy.pdparams +3 -0
- best_accuracy.states +0 -0
- config.yml +151 -0
.gitattributes
CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
+
best_accuracy.pdopt filter=lfs diff=lfs merge=lfs -text
|
37 |
+
best_accuracy.pdparams filter=lfs diff=lfs merge=lfs -text
|
best_accuracy.pdopt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:397bf7dc4e9443b43a44412c227aec2d8794e311ec23bfc8b02049418ec6ec18
|
3 |
+
size 111245538
|
best_accuracy.pdparams
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b952f6898313a9e1dda4b9e36b881ddcaf583357c10195f0df49b005746526fe
|
3 |
+
size 89287511
|
best_accuracy.states
ADDED
Binary file (174 Bytes). View file
|
|
config.yml
ADDED
@@ -0,0 +1,151 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
Global:
|
2 |
+
debug: false
|
3 |
+
use_gpu: true
|
4 |
+
epoch_num: 25
|
5 |
+
log_smooth_window: 20
|
6 |
+
print_batch_step: 50
|
7 |
+
save_model_dir: ./output/final_e25
|
8 |
+
save_epoch_step: 14
|
9 |
+
eval_batch_step:
|
10 |
+
- 0
|
11 |
+
- 697
|
12 |
+
cal_metric_during_train: true
|
13 |
+
pretrained_model: ./models/latin_PP-OCRv3_rec_train/best_accuracy
|
14 |
+
checkpoints: null
|
15 |
+
save_inference_dir: ./inference/final
|
16 |
+
use_visualdl: false
|
17 |
+
infer_img: doc/imgs_words/ch/word_1.jpg
|
18 |
+
character_dict_path: ppocr/utils/dict/latin_dict.txt
|
19 |
+
max_text_length: 25
|
20 |
+
infer_mode: false
|
21 |
+
use_space_char: true
|
22 |
+
distributed: true
|
23 |
+
save_res_path: ./output/rec/predicts_ppocrv3.txt
|
24 |
+
seed: 42
|
25 |
+
Optimizer:
|
26 |
+
name: Adam
|
27 |
+
beta1: 0.9
|
28 |
+
beta2: 0.999
|
29 |
+
lr:
|
30 |
+
name: Cosine
|
31 |
+
learning_rate: 0.001
|
32 |
+
warmup_epoch: 5
|
33 |
+
regularizer:
|
34 |
+
name: L2
|
35 |
+
factor: 3.0e-05
|
36 |
+
Architecture:
|
37 |
+
model_type: rec
|
38 |
+
algorithm: SVTR
|
39 |
+
Transform: null
|
40 |
+
Backbone:
|
41 |
+
name: MobileNetV1Enhance
|
42 |
+
scale: 0.5
|
43 |
+
last_conv_stride:
|
44 |
+
- 1
|
45 |
+
- 2
|
46 |
+
last_pool_type: avg
|
47 |
+
Head:
|
48 |
+
name: MultiHead
|
49 |
+
head_list:
|
50 |
+
- CTCHead:
|
51 |
+
Neck:
|
52 |
+
name: svtr
|
53 |
+
dims: 64
|
54 |
+
depth: 2
|
55 |
+
hidden_dims: 120
|
56 |
+
use_guide: true
|
57 |
+
Head:
|
58 |
+
fc_decay: 1.0e-05
|
59 |
+
- SARHead:
|
60 |
+
enc_dim: 512
|
61 |
+
max_text_length: 25
|
62 |
+
Loss:
|
63 |
+
name: MultiLoss
|
64 |
+
loss_config_list:
|
65 |
+
- CTCLoss: null
|
66 |
+
- SARLoss: null
|
67 |
+
PostProcess:
|
68 |
+
name: CTCLabelDecode
|
69 |
+
Metric:
|
70 |
+
name: RecMetric
|
71 |
+
main_indicator: acc
|
72 |
+
keep_space: false
|
73 |
+
Train:
|
74 |
+
dataset:
|
75 |
+
name: SimpleDataSet
|
76 |
+
data_dir: /data/AIDA/iter4/rivikuvat
|
77 |
+
label_file_list:
|
78 |
+
- /data/AIDA/iter4/annot/train_kone_best_iter2_14_02_24.txt
|
79 |
+
- /data/AIDA/iter4/annot/train_kone_best_iter3_04_03_24.txt
|
80 |
+
- /data/AIDA/iter4/annot/train_sini3_2_19_03_kone_best.txt
|
81 |
+
- /data/AIDA/iter4/annot/train_kasi_best_iter2_14_2_24.txt
|
82 |
+
- /data/AIDA/iter4/annot/train_kasi_best_iter3_04_03_24.txt
|
83 |
+
- /data/AIDA/iter4/annot/train_sini3_2_19_03_kasi_best.txt
|
84 |
+
- /data/AIDA/iter4/annot/normaali_16k_train.txt
|
85 |
+
- /data/AIDA/iter4/annot/shaken_train.txt
|
86 |
+
- /data/AIDA/iter4/annot/noise_k5_train.txt
|
87 |
+
- /data/AIDA/iter4/annot/noise_k5_sin_train.txt
|
88 |
+
transforms:
|
89 |
+
- DecodeImage:
|
90 |
+
img_mode: BGR
|
91 |
+
channel_first: false
|
92 |
+
- RecAug:
|
93 |
+
reverse_prob: 0.25
|
94 |
+
noise_prob: 0.25
|
95 |
+
blur_prob: 0.25
|
96 |
+
hsv_aug_prob: 0.25
|
97 |
+
- MultiLabelEncode: null
|
98 |
+
- SARRecResizeImg:
|
99 |
+
image_shape:
|
100 |
+
- 3
|
101 |
+
- 48
|
102 |
+
- 48
|
103 |
+
- 320
|
104 |
+
width_downsample_ratio: 0.25
|
105 |
+
mode: v3
|
106 |
+
- KeepKeys:
|
107 |
+
keep_keys:
|
108 |
+
- image
|
109 |
+
- label_ctc
|
110 |
+
- label_sar
|
111 |
+
- length
|
112 |
+
- valid_ratio
|
113 |
+
loader:
|
114 |
+
shuffle: true
|
115 |
+
batch_size_per_card: 128
|
116 |
+
drop_last: true
|
117 |
+
num_workers: 22
|
118 |
+
Eval:
|
119 |
+
dataset:
|
120 |
+
name: SimpleDataSet
|
121 |
+
data_dir: /data/AIDA/iter4/rivikuvat
|
122 |
+
label_file_list:
|
123 |
+
- /data/AIDA/iter4/annot/valid_kone_best_iter2_filttered_14_2_24.txt
|
124 |
+
- /data/AIDA/iter4/annot/valid_kone_best_iter3_04_03_24.txt
|
125 |
+
- /data/AIDA/iter4/annot/valid_sini3_2_19_03_kone_best.txt
|
126 |
+
transforms:
|
127 |
+
- DecodeImage:
|
128 |
+
img_mode: BGR
|
129 |
+
channel_first: false
|
130 |
+
- MultiLabelEncode: null
|
131 |
+
- SARRecResizeImg:
|
132 |
+
image_shape:
|
133 |
+
- 3
|
134 |
+
- 48
|
135 |
+
- 48
|
136 |
+
- 320
|
137 |
+
width_downsample_ratio: 0.25
|
138 |
+
mode: v3
|
139 |
+
- KeepKeys:
|
140 |
+
keep_keys:
|
141 |
+
- image
|
142 |
+
- label_ctc
|
143 |
+
- label_sar
|
144 |
+
- length
|
145 |
+
- valid_ratio
|
146 |
+
loader:
|
147 |
+
shuffle: false
|
148 |
+
drop_last: false
|
149 |
+
batch_size_per_card: 128
|
150 |
+
num_workers: 16
|
151 |
+
profiler_options: null
|