Stanislav Kalinin commited on
Commit
d9c5d39
1 Parent(s): 856d9bd

feat: Update OCR models

Browse files

feat: Add OCR model onnx

Files changed (3) hide show
  1. ocr/ocr_config.json +60 -0
  2. ocr/ocr_model.ckpt +3 -0
  3. ocr/ocr_model.onnx +3 -0
ocr/ocr_config.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alphabet": " #()+/0123456789[]abdefghiklmnoprstuw|×ǂабвгдежзийклмнопрстуфхцчшщъыьэюяѣ–⊕⊗",
3
+ "save_dir": "/home/jovyan/skalinin/OCR-model-replace-layer/OCR-model/data/experiments_peter/change_ocr_layer_for_ONNX_2",
4
+ "num_epochs": 200,
5
+ "pretrain_path": "/home/jovyan/skalinin/OCR-model-replace-layer/OCR-model/data/peter_pretrain/ocr_model.ckpt",
6
+ "image": {
7
+ "width": 1024,
8
+ "height": 128
9
+ },
10
+ "train": {
11
+ "datasets": [
12
+ {
13
+ "csv_path": "/home/jovyan/data/ocr/peter_mask/train.csv",
14
+ "prob": 10
15
+ },
16
+ {
17
+ "csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/generated_synth/stmix-peter-synth-mask-part1.csv",
18
+ "prob": 20
19
+ },
20
+ {
21
+ "csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/generated_synth/stmix-peter-synth-mask-part2.csv",
22
+ "prob": 20
23
+ },
24
+ {
25
+ "csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/generated_synth/stmix-peter-synth-part-4-fix.csv",
26
+ "prob": 10
27
+ },
28
+ {
29
+ "csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/generated_synth/stmix-peter-synth-part-5-fix.csv",
30
+ "prob": 10
31
+ },
32
+ {
33
+ "csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/peter_mask_crops/peter_mask/stackmix_crops_ocr_format.csv",
34
+ "prob": 10
35
+ }
36
+ ],
37
+ "epoch_size": 40000,
38
+ "batch_size": 128
39
+ },
40
+ "val": {
41
+ "datasets": [
42
+ {
43
+ "csv_path": "/home/jovyan/data/ocr/peter_mask/val.csv",
44
+ "prob": 1
45
+ }
46
+ ],
47
+ "epoch_size": null,
48
+ "batch_size": 128
49
+ },
50
+ "test": {
51
+ "datasets": [
52
+ {
53
+ "csv_path": "/home/jovyan/data/ocr/peter_mask/test.csv",
54
+ "prob": 1
55
+ }
56
+ ],
57
+ "epoch_size": null,
58
+ "batch_size": 128
59
+ }
60
+ }
ocr/ocr_model.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2770258153a021be6fb29bb9ea067f1c745a7699d620926b0e2612a56b7ba4b7
3
+ size 50217625
ocr/ocr_model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:799a035b501b10bab528bd89a2477ec491d771454e920c6531f68246662ba493
3
+ size 50103685