feat: Add weights and configs for Peter models

ocr/model-95-0.0293.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4a8cc37c8483b7d33dc5fecbcd0afabfeb48192c64db56dc11d16225efc93d40
+size 50222851

ocr/ocr_config.json

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

pipeline_config.json

@@ -0,0 +1,39 @@
+{   "main_process": {
+        "ImageToBGR": {
+            "input_format": "BGR"
+        },
+        "SegmPrediction": {
+            "model_path": "segm/model-100-0.1701.ckpt",
+            "config_path": "segm/segm_config.json",
+            "device": "cuda"
+        },
+        "RestoreImageAngle": {
+            "restoring_class_names": ["shrinked_text"]
+        },
+        "ClassContourPosptrocess": {},
+        "OCRPrediction": {
+            "model_path": "ocr/model-95-0.0293.ckpt",
+            "config_path": "ocr/ocr_config.json",
+            "lm_path": "",
+            "classes_to_ocr": ["shrinked_text"],
+            "device": "cuda"
+        },
+        "LineFinder": {
+            "line_classes": ["shrinked_text"],
+            "text_classes": ["shrinked_text"],
+            "pages_clust_dist": 0.35
+        },
+        "PrepareJSON": {}
+    },
+    "classes": {
+        "shrinked_text": {
+            "contour_posptrocess": {
+                "UpscaleContour": {"upscale_contour": 1.45},
+                "BboxFromContour": {},
+                "CropByBbox": {},
+                "MakeMaskedCrop": {},
+                "RotateVerticalCrops": {}
+            }
+        }
+    }
+}

segm/model-100-0.1701.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:137ea5f4eb979c2ddece70b5777da840afa139c2fc36a81b83c019431c62c896
+size 115723399

segm/segm_config.json

@@ -0,0 +1,67 @@
+{
+    "save_dir": "/home/jovyan/skalinin/SEGM-model/data/exp_peter/segm_baseline_pretrain_2",
+    "num_epochs": 1500,
+    "pretrain_path": "/home/jovyan/skalinin/SEGM-model/data/exp/resnet50_896_896_border_all_text_thin_border/model-1404-0.1957.ckpt",
+    "image": {
+        "width": 896,
+        "height": 896
+    },
+    "classes": {
+        "shrinked_text": {
+            "annotation_classes": ["text"],
+            "polygon2mask": {
+                "ShrinkMaskMaker": {"shrink_ratio": 0.75}
+            },
+            "postprocess": {
+                "threshold": 0.7,
+                "min_area": 100
+            }
+        },
+        "bordered_text": {
+            "annotation_classes": ["text"],
+            "polygon2mask": {
+                "BorderMaskMaker": {"shrink_ratio": 0.5}
+            },
+            "postprocess": {
+                "threshold": 0.7,
+                "min_area": 100
+            }
+        }
+    },
+    "train": {
+        "datasets": [
+            {
+                "json_path": "/home/jovyan/data/segm/peter/annotations_train.json",
+                "image_root": "/home/jovyan/data/segm/peter/images/",
+                "processed_data_path": "/home/jovyan/skalinin/SEGM-model/data/peter/annotations_train.csv",
+                "prob": 0.1
+            }
+        ],
+        "epoch_size": 2000,
+        "batch_size": 10
+    },
+    "val": {
+        "datasets": [
+            {
+                "json_path": "/home/jovyan/data/segm/peter/annotations_val.json",
+                "image_root": "/home/jovyan/data/segm/peter/images/",
+                "processed_data_path": "/home/jovyan/skalinin/SEGM-model/data/peter/annotations_val.csv",
+                "prob": 1
+            }
+        ],
+        "epoch_size": null,
+        "batch_size": 10
+    },
+    "test": {
+        "datasets": [
+            {
+                "json_path": "/home/jovyan/data/segm/peter/annotations_test.json",
+                "image_root": "/home/jovyan/data/segm/peter/images/",
+                "processed_data_path": "/home/jovyan/skalinin/SEGM-model/data/peter/annotations_test.csv",
+                "prob": 1
+            }
+        ],
+        "epoch_size": null,
+        "batch_size": 10
+    }
+}