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