fohra commited on
Commit
edf9459
1 Parent(s): 0a0d6cd

Upload model weights for training

Browse files
.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