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