eynollah/train/config_params.json

43 lines
1.2 KiB
JSON
Raw Normal View History

{
2024-05-07 13:34:03 +02:00
"backbone_type" : "nontransformer",
"task": "classification",
"n_classes" : 2,
"n_epochs" : 20,
2024-05-06 18:31:48 +02:00
"input_height" : 448,
"input_width" : 448,
"weight_decay" : 1e-6,
2024-05-07 13:34:03 +02:00
"n_batch" : 6,
"learning_rate": 1e-4,
"f1_threshold_classification": 0.8,
2024-05-06 18:31:48 +02:00
"patches" : true,
"pretraining" : true,
2021-06-22 18:47:59 -04:00
"augmentation" : false,
"flip_aug" : false,
2021-06-22 18:47:59 -04:00
"blur_aug" : false,
"scaling" : true,
"binarization" : false,
"scaling_bluring" : false,
"scaling_binarization" : false,
2021-06-22 14:20:51 -04:00
"scaling_flip" : false,
"rotation": false,
2021-06-22 14:20:51 -04:00
"rotation_not_90": false,
2024-05-07 13:34:03 +02:00
"transformer_num_patches_xy": [28, 28],
2024-04-16 01:00:48 +02:00
"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],
2024-05-07 13:34:03 +02:00
"classification_classes_name" : {"0":"apple", "1":"orange"},
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-05-07 13:34:03 +02:00
"dir_train": "./train",
2024-05-06 18:31:48 +02:00
"dir_eval": "./eval",
2024-05-07 13:34:03 +02:00
"dir_output": "./output"
}