{ "save_dir": "/home/jovyan/skalinin/SEGM-model/data/exp_peter/segm_baseline_pretrain_2", "num_epochs": 1500, "pretrain_path": "/home/jovyan/skalinin/SEGM-model/data/exp/resnet50_896_896_border_all_text_thin_border/model-1404-0.1957.ckpt", "image": { "width": 896, "height": 896 }, "classes": { "shrinked_text": { "annotation_classes": ["text"], "polygon2mask": { "ShrinkMaskMaker": {"shrink_ratio": 0.75} }, "postprocess": { "threshold": 0.7, "min_area": 100 } }, "bordered_text": { "annotation_classes": ["text"], "polygon2mask": { "BorderMaskMaker": {"shrink_ratio": 0.5} }, "postprocess": { "threshold": 0.7, "min_area": 100 } } }, "train": { "datasets": [ { "json_path": "/home/jovyan/data/segm/peter/annotations_train.json", "image_root": "/home/jovyan/data/segm/peter/images/", "processed_data_path": "/home/jovyan/skalinin/SEGM-model/data/peter/annotations_train.csv", "prob": 0.1 } ], "epoch_size": 2000, "batch_size": 10 }, "val": { "datasets": [ { "json_path": "/home/jovyan/data/segm/peter/annotations_val.json", "image_root": "/home/jovyan/data/segm/peter/images/", "processed_data_path": "/home/jovyan/skalinin/SEGM-model/data/peter/annotations_val.csv", "prob": 1 } ], "epoch_size": null, "batch_size": 10 }, "test": { "datasets": [ { "json_path": "/home/jovyan/data/segm/peter/annotations_test.json", "image_root": "/home/jovyan/data/segm/peter/images/", "processed_data_path": "/home/jovyan/skalinin/SEGM-model/data/peter/annotations_test.csv", "prob": 1 } ], "epoch_size": null, "batch_size": 10 } }