File size: 3,440 Bytes
9c0ac7a ce41b2a 76f1094 ce41b2a 9c0ac7a ce41b2a 9c0ac7a ce41b2a 9c0ac7a ce41b2a 9c0ac7a ce41b2a 9c0ac7a ce41b2a 9c0ac7a ce41b2a 9c0ac7a 76f1094 ce41b2a 9c0ac7a ce41b2a 9c0ac7a ce41b2a 9c0ac7a ce41b2a 9c0ac7a ce41b2a 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 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 |
{
"save_dir": "data/experiments/tetradi_with_lines/",
"num_epochs": 2000,
"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": 1500,
"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
}
}
|