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

adding enhancement training

Browse source
This commit is contained in:
vahidrezanezhad 2024-05-06 18:31:48 +02:00
parent dbb84507ed
commit 38db3e9289
5 changed files with 119 additions and 68 deletions
Download patch file Download diff file Expand all files Collapse all files

20
config_params.json
View file

@ -1,15 +1,15 @@
{
"model_name" : "resnet50_unet",
"task": "classification",
"n_classes" : 2,
"n_epochs" : 7,
"input_height" : 224,
"input_width" : 224,
"task": "enhancement",
"n_classes" : 3,
"n_epochs" : 3,
"input_height" : 448,
"input_width" : 448,
"weight_decay" : 1e-6,
"n_batch" : 6,
"n_batch" : 3,
"learning_rate": 1e-4,
"f1_threshold_classification": 0.8,
"patches" : false,
"patches" : true,
"pretraining" : true,
"augmentation" : false,
"flip_aug" : false,
@ -35,7 +35,7 @@
"weighted_loss": false,
"is_loss_soft_dice": false,
"data_is_provided": false,
"dir_train": "/home/vahid/Downloads/image_classification_data/train",
"dir_eval": "/home/vahid/Downloads/image_classification_data/eval",
"dir_output": "/home/vahid/Downloads/image_classification_data/output"
"dir_train": "./training_data_sample_enhancement",
"dir_eval": "./eval",
"dir_output": "./out"
}