eynollah/train/config_params.json

{
    "model_name" : "resnet50_unet",
    "task": "enhancement",
    "n_classes" : 3,
    "n_epochs" : 3,
    "input_height" : 448,
    "input_width" : 448,
    "weight_decay" : 1e-6,
    "n_batch" : 3,
    "learning_rate": 1e-4,
    "f1_threshold_classification": 0.8,
    "patches" : true,
    "pretraining" : true,
    "augmentation" : false,
    "flip_aug" : false,
    "blur_aug" : false,
    "scaling" : true,
    "binarization" : false,
    "scaling_bluring" : false,
    "scaling_binarization" : false,
    "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" : " ", 
    "weighted_loss": false,
    "is_loss_soft_dice": false,
    "data_is_provided": false,
    "dir_train": "./training_data_sample_enhancement",
    "dir_eval": "./eval",
    "dir_output": "./out"
}