mirror of
https://github.com/qurator-spk/sbb_pixelwise_segmentation.git
synced 2025-07-01 14:40:05 +02:00
55 lines
1.6 KiB
JSON
55 lines
1.6 KiB
JSON
|
{
|
||
|
|
"backbone_type" : "nontransformer",
|
||
|
|
"task": "segmentation",
|
||
|
|
"n_classes" : 3,
|
||
|
|
"n_epochs" : 1,
|
||
|
|
"input_height" : 672,
|
||
|
|
"input_width" : 448,
|
||
|
|
"weight_decay" : 1e-6,
|
||
|
|
"n_batch" : 4,
|
||
|
|
"learning_rate": 1e-4,
|
||
|
|
"patches" : false,
|
||
|
|
"pretraining" : true,
|
||
|
|
"augmentation" : false,
|
||
|
|
"flip_aug" : false,
|
||
|
|
"blur_aug" : true,
|
||
|
|
"scaling" : true,
|
||
|
|
"adding_rgb_background": false,
|
||
|
|
"adding_rgb_foreground": false,
|
||
|
|
"add_red_textlines": false,
|
||
|
|
"channels_shuffling": true,
|
||
|
|
"degrading": true,
|
||
|
|
"brightening": true,
|
||
|
|
"binarization" : false,
|
||
|
|
"scaling_bluring" : false,
|
||
|
|
"scaling_binarization" : false,
|
||
|
|
"scaling_flip" : false,
|
||
|
|
"rotation": false,
|
||
|
|
"rotation_not_90": true,
|
||
|
|
"transformer_num_patches_xy": [14, 21],
|
||
|
|
"transformer_patchsize_x": 1,
|
||
|
|
"transformer_patchsize_y": 1,
|
||
|
|
"transformer_projection_dim": 64,
|
||
|
|
"transformer_mlp_head_units": [128, 64],
|
||
|
|
"transformer_layers": 1,
|
||
|
|
"transformer_num_heads": 1,
|
||
|
|
"transformer_cnn_first": true,
|
||
|
|
"blur_k" : ["blur","gauss","median"],
|
||
|
|
"scales" : [0.6, 0.7, 0.8, 0.9],
|
||
|
|
"brightness" : [1.3, 1.5, 1.7, 2],
|
||
|
|
"degrade_scales" : [0.2, 0.4],
|
||
|
|
"flip_index" : [0, 1, -1],
|
||
|
|
"shuffle_indexes" : [ [0,2,1], [1,2,0], [1,0,2] , [2,1,0]],
|
||
|
|
"thetha" : [5, -5],
|
||
|
|
"number_of_backgrounds_per_image": 2,
|
||
|
|
"continue_training": false,
|
||
|
|
"index_start" : 0,
|
||
|
|
"dir_of_start_model" : " ",
|
||
|
|
"weighted_loss": false,
|
||
|
|
"is_loss_soft_dice": true,
|
||
|
|
"data_is_provided": false,
|
||
|
|
"dir_train": "/entry_point_dir/train",
|
||
|
|
"dir_eval": "/entry_point_dir/eval",
|
||
|
|
"dir_output": "/entry_point_dir/output"
|
||
|
|
}
|