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

scaling, channels shuffling, rgb background and red content added to no patch augmentation

Browse source
This commit is contained in:
vahidrezanezhad 2024-08-21 19:33:23 +02:00
parent 95bbdf8040
commit f31219b1c9
3 changed files with 68 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

30
config_params.json
View file

@ -1,19 +1,22 @@
{
"backbone_type" : "transformer",
"task": "binarization",
"task": "segmentation",
"n_classes" : 2,
"n_epochs" : 2,
"input_height" : 224,
"input_width" : 224,
"n_epochs" : 0,
"input_height" : 448,
"input_width" : 448,
"weight_decay" : 1e-6,
"n_batch" : 1,
"learning_rate": 1e-4,
"patches" : true,
"patches" : false,
"pretraining" : true,
"augmentation" : false,
"augmentation" : true,
"flip_aug" : false,
"blur_aug" : false,
"scaling" : true,
"adding_rgb_background": true,
"add_red_textlines": true,
"channels_shuffling": true,
"degrading": false,
"brightening": false,
"binarization" : false,
@ -31,18 +34,23 @@
"transformer_num_heads": 1,
"transformer_cnn_first": false,
"blur_k" : ["blur","guass","median"],
"scales" : [0.6, 0.7, 0.8, 0.9, 1.1, 1.2, 1.4],
"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],
"thetha" : [10, -10],
"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": false,
"data_is_provided": false,
"dir_train": "./train",
"dir_eval": "./eval",
"dir_output": "./output"
"dir_train": "/home/vahid/Documents/test/sbb_pixelwise_segmentation/test_label/pageextractor_test/train_new",
"dir_eval": "/home/vahid/Documents/test/sbb_pixelwise_segmentation/test_label/pageextractor_test/eval_new",
"dir_output": "/home/vahid/Documents/test/sbb_pixelwise_segmentation/test_label/pageextractor_test/output_new",
"dir_rgb_backgrounds": "/home/vahid/Documents/1_2_test_eynollah/set_rgb_background",
"dir_img_bin": "/home/vahid/Documents/test/sbb_pixelwise_segmentation/test_label/pageextractor_test/train_new/images_bin"
}