qurator-spk/sbb_pixelwise_segmentation
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_pixelwise_segmentation.git synced 2025-06-09 20:00:05 +02:00
Code Issues Releases Wiki Activity

first working update of branch

Browse source
This commit is contained in:
vahidrezanezhad 2024-04-16 01:00:48 +02:00
parent 02b1436f39
commit d27647a0f1
4 changed files with 452 additions and 151 deletions
Download patch file Download diff file Expand all files Collapse all files

19
config_params.json
View file

@ -1,8 +1,9 @@
{
"n_classes" : 3,
"model_name" : "hybrid_transformer_cnn",
"n_classes" : 2,
"n_epochs" : 2,
"input_height" : 448,
"input_width" : 672,
"input_width" : 448,
"weight_decay" : 1e-6,
"n_batch" : 2,
"learning_rate": 1e-4,
@ -18,13 +19,21 @@
"scaling_flip" : false,
"rotation": false,
"rotation_not_90": false,
"num_patches_xy": [28, 28],
"transformer_patchsize": 1,
"blur_k" : ["blur","guass","median"],
"scales" : [0.6, 0.7, 0.8, 0.9, 1.1, 1.2, 1.4],
"brightness" : [1.3, 1.5, 1.7, 2],
"degrade_scales" : [0.2, 0.4],
"flip_index" : [0, 1, -1],
"thetha" : [10, -10],
"continue_training": false,
"index_start": 0,
"dir_of_start_model": " ",
"index_start" : 0,
"dir_of_start_model" : " ",
"weighted_loss": false,
"is_loss_soft_dice": false,
"data_is_provided": false,
"dir_train": "/train",
"dir_eval": "/eval",
"dir_output": "/output"
"dir_output": "/out"
}