feat: Update OCR models

feat: Add OCR model onnx

ocr/ocr_config.json

@@ -0,0 +1,60 @@
+{
+    "alphabet": " #()+/0123456789[]abdefghiklmnoprstuw|×ǂабвгдежзийклмнопрстуфхцчшщъыьэюяѣ–⊕⊗",
+    "save_dir": "/home/jovyan/skalinin/OCR-model-replace-layer/OCR-model/data/experiments_peter/change_ocr_layer_for_ONNX_2",
+    "num_epochs": 200,
+    "pretrain_path": "/home/jovyan/skalinin/OCR-model-replace-layer/OCR-model/data/peter_pretrain/ocr_model.ckpt",
+    "image": {
+        "width": 1024,
+        "height": 128
+    },
+    "train": {
+        "datasets": [
+            {
+                "csv_path": "/home/jovyan/data/ocr/peter_mask/train.csv",
+                "prob": 10
+            },
+            {
+                "csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/generated_synth/stmix-peter-synth-mask-part1.csv",
+                "prob": 20
+            },
+            {
+                "csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/generated_synth/stmix-peter-synth-mask-part2.csv",
+                "prob": 20
+            },
+            {
+                "csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/generated_synth/stmix-peter-synth-part-4-fix.csv",
+                "prob": 10
+            },
+            {
+                "csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/generated_synth/stmix-peter-synth-part-5-fix.csv",
+                "prob": 10
+            },
+            {
+                "csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/peter_mask_crops/peter_mask/stackmix_crops_ocr_format.csv",
+                "prob": 10
+            }
+        ],
+        "epoch_size": 40000,
+        "batch_size": 128
+    },
+    "val": {
+        "datasets": [
+            {
+                "csv_path": "/home/jovyan/data/ocr/peter_mask/val.csv",
+                "prob": 1
+            }
+        ],
+        "epoch_size": null,
+        "batch_size": 128
+    },
+    "test": {
+        "datasets": [
+            {
+                "csv_path": "/home/jovyan/data/ocr/peter_mask/test.csv",
+                "prob": 1
+            }
+        ],
+        "epoch_size": null,
+        "batch_size": 128
+    }
+}

ocr/ocr_model.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2770258153a021be6fb29bb9ea067f1c745a7699d620926b0e2612a56b7ba4b7
+size 50217625

ocr/ocr_model.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:799a035b501b10bab528bd89a2477ec491d771454e920c6531f68246662ba493
+size 50103685