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