2019-12-09 15:33:53 +01:00
|
|
|
{
|
2024-04-29 20:59:36 +02:00
|
|
|
"model_name" : "resnet50_unet",
|
|
|
|
|
"task": "classification",
|
2024-04-16 01:00:48 +02:00
|
|
|
"n_classes" : 2,
|
2024-04-29 20:59:36 +02:00
|
|
|
"n_epochs" : 7,
|
|
|
|
|
"input_height" : 224,
|
|
|
|
|
"input_width" : 224,
|
2019-12-09 15:33:53 +01:00
|
|
|
"weight_decay" : 1e-6,
|
2024-04-29 20:59:36 +02:00
|
|
|
"n_batch" : 6,
|
2019-12-09 15:33:53 +01:00
|
|
|
"learning_rate": 1e-4,
|
2024-04-29 20:59:36 +02:00
|
|
|
"f1_threshold_classification": 0.8,
|
|
|
|
|
"patches" : false,
|
2019-12-09 15:33:53 +01:00
|
|
|
"pretraining" : true,
|
2021-06-22 18:47:59 -04:00
|
|
|
"augmentation" : false,
|
2019-12-09 15:33:53 +01:00
|
|
|
"flip_aug" : false,
|
2021-06-22 18:47:59 -04:00
|
|
|
"blur_aug" : false,
|
|
|
|
|
"scaling" : true,
|
2019-12-09 15:33:53 +01:00
|
|
|
"binarization" : false,
|
|
|
|
|
"scaling_bluring" : false,
|
|
|
|
|
"scaling_binarization" : false,
|
2021-06-22 14:20:51 -04:00
|
|
|
"scaling_flip" : false,
|
2019-12-09 15:33:53 +01:00
|
|
|
"rotation": false,
|
2021-06-22 14:20:51 -04:00
|
|
|
"rotation_not_90": false,
|
2024-04-16 01:00:48 +02:00
|
|
|
"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],
|
2021-06-22 18:47:59 -04:00
|
|
|
"continue_training": false,
|
2024-04-16 01:00:48 +02:00
|
|
|
"index_start" : 0,
|
|
|
|
|
"dir_of_start_model" : " ",
|
2021-06-22 18:47:59 -04:00
|
|
|
"weighted_loss": false,
|
|
|
|
|
"is_loss_soft_dice": false,
|
|
|
|
|
"data_is_provided": false,
|
2024-04-29 20:59:36 +02:00
|
|
|
"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"
|
2019-12-09 15:33:53 +01:00
|
|
|
}
|
