File size: 2,606 Bytes
9c0ac7a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
{
"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
}
}
|