File size: 2,194 Bytes
54c5290 |
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 |
{
"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
}
}
|