{
    "save_dir": "data/experiments/tetradi_with_lines/",
    "num_epochs": 2000,
    "pretrain_path": "data/experiments/base_model_all_datasets/model-960-0.1979.ckpt",
    "image": {
        "width": 896,
        "height": 896
    },
    "classes": {
        "shrinked_text": {
            "annotation_classes": ["pupil_text", "teacher_comment"],
            "polygon2mask": {
                "ShrinkMaskMaker": {"shrink_ratio": 0.5}
            },
            "postprocess": {
                "threshold": 0.8,
                "min_area": 10
            }
        },
        "bordered_text": {
            "annotation_classes": ["pupil_text", "teacher_comment"],
            "polygon2mask": {
                "BorderMaskMaker": {"shrink_ratio": 0.5}
            },
            "postprocess": {
                "threshold": 0.8,
                "min_area": 10
            }
        },
        "text_line": {
            "annotation_classes": ["text_line"],
            "polygon2mask": {
                "PolylineToMask": {"thickness": 2}
            },
            "postprocess": {
                "threshold": 0.8,
                "min_area": 10
            }
        }
    },
    "train": {
        "datasets": [
            {
                "json_path": "data/segm/school_notebooks_EN/annotations_train.json",
                "image_root": "data/segm/school_notebooks_EN/images/",
                "processed_data_path": "data/tetradi_with_lines/school_notebooks_EN/train/annotations_train.csv",
                "prob": 70
            },
            {
                "json_path": "data/segm/school_notebooks_RU/annotations_train.json",
                "image_root": "data/segm/school_notebooks_RU/images/",
                "processed_data_path": "data/tetradi_with_lines/school_notebooks_RU/train/annotations_train.csv",
                "prob": 1557
            }
        ],
        "epoch_size": 1500,
        "batch_size": 20
    },
    "val": {
        "datasets": [
            {
                "json_path": "data/segm/school_notebooks_EN/annotations_val.json",
                "image_root": "data/segm/school_notebooks_EN/images/",
                "processed_data_path": "data/tetradi_with_lines/school_notebooks_EN/val/annotations_val.csv",
                "prob": 10
            },
            {
                "json_path": "data/segm/school_notebooks_RU/annotations_val.json",
                "image_root": "data/segm/school_notebooks_RU/images/",
                "processed_data_path": "data/tetradi_with_lines/school_notebooks_RU/val/annotations_val.csv",
                "prob": 150
            }
        ],
        "epoch_size": null,
        "batch_size": 20
    },
    "test": {
        "datasets": [
            {
                "json_path": "data/segm/school_notebooks_EN/annotations_test.json",
                "image_root": "data/segm/school_notebooks_EN/images/",
                "processed_data_path": "data/tetradi_with_lines/school_notebooks_EN/test/annotations_test.csv",
                "prob": 10
            },
            {
                "json_path": "data/segm/school_notebooks_RU/annotations_test.json",
                "image_root": "data/segm/school_notebooks_RU/images/",
                "processed_data_path": "data/tetradi_with_lines/school_notebooks_RU/test/annotations_test.csv",
                "prob": 150
            }
        ],
        "epoch_size": null,
        "batch_size": 20
    }
}