{ "save_dir": "/home/jovyan/SEGM-model/data/exp/fitting_text_line_11032022", "num_epochs": 1000, "pretrain_path": "/home/jovyan/OCR-pipeline/data/weights/resnet50_15102021_26012022_768_1024_2/model-1249-0.1312.ckpt", "image": { "width": 1024, "height": 768 }, "classes": { "shrinked_pupil_text": { "annotation_classes": ["pupil_text"], "polygon2mask": { "ShrinkMaskMaker": {"shrink_ratio": 0.5} }, "postprocess": { "threshold": 0.8, "min_area": 10 } }, "shrinked_comments": { "annotation_classes": ["pupil_comment", "teacher_comment"], "polygon2mask": { "ShrinkMaskMaker": {"shrink_ratio": 0.5} }, "postprocess": { "threshold": 0.8, "min_area": 10 } }, "text_line": { "annotation_classes": ["text_line"], "polygon2mask": { "PolylineToMask": {"thickness": 2, "scale": 0.95} }, "postprocess": { "threshold": 0.8, "min_area": 10 } } }, "train": { "datasets": [ { "json_path": "/home/jovyan/data/htr_dataset_SEGM_11032022/en_ru_tetradi/ru/annotations_train.json", "image_root": "/home/jovyan/data/htr_dataset_SEGM_11032022/en_ru_tetradi/ru/images_rotated/", "processed_data_path": "/home/jovyan/SEGM-model/data/htr_dataset_SEGM_11032022_ru/annotations_train.csv" } ], "batch_size": 10 }, "val": { "datasets": [ { "json_path": "/home/jovyan/data/htr_dataset_SEGM_11032022/en_ru_tetradi/ru/annotations_val.json", "image_root": "/home/jovyan/data/htr_dataset_SEGM_11032022/en_ru_tetradi/ru/images_rotated/", "processed_data_path": "/home/jovyan/SEGM-model/data/htr_dataset_SEGM_11032022_ru/annotations_val.csv" } ], "batch_size": 10 }, "test": { "datasets": [ { "json_path": "/home/jovyan/data/htr_dataset_SEGM_11032022/en_ru_tetradi/ru/annotations_test.json", "image_root": "/home/jovyan/data/htr_dataset_SEGM_11032022/en_ru_tetradi/ru/images_rotated/", "processed_data_path": "/home/jovyan/SEGM-model/data/htr_dataset_SEGM_11032022_ru/annotations_val.csv" } ], "batch_size": 10 } }