| { | |
| "save_dir": "data/experiments/tetradi_with_lines/", | |
| "num_epochs": 1500, | |
| "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": 2000, | |
| "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 | |
| } | |
| } | |