| { | |
| "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 | |
| } | |
| } | |