a b/settings.ini
1
[COMMON]
2
save_model_dir = "saved_models"
3
model_name = "quicknat"
4
log_dir = "logs"
5
device = 1
6
exp_dir = "experiments"
7
8
[DATA]
9
data_dir = "datasets/COMBINED/axial"
10
train_data_file = "Data_train.h5"
11
train_label_file = "Label_train.h5"
12
train_class_weights_file = "Class_Weight_train.h5"
13
train_weights_file = "Weight_train.h5"
14
test_data_file = "Data_test.h5"
15
test_label_file = "Label_test.h5"
16
test_class_weights_file = "Class_Weight_test.h5"
17
test_weights_file = "Weight_test.h5"
18
labels = ["Background", "Left WM", "Left Cortex", "Left Lateral ventricle", "Left Inf LatVentricle", "Left Cerebellum WM", "Left Cerebellum Cortex", "Left Thalamus", "Left Caudate", "Left Putamen", "Left Pallidum", "3rd Ventricle", "4th Ventricle", "Brain Stem", "Left Hippocampus", "Left Amygdala", "CSF (Cranial)", "Left Accumbens", "Left Ventral DC", "Right WM", "Right Cortex", "Right Lateral Ventricle", "Right Inf LatVentricle", "Right Cerebellum WM", "Right Cerebellum Cortex", "Right Thalamus", "Right Caudate", "Right Putamen", "Right Pallidum", "Right Hippocampus", "Right Amygdala", "Right Accumbens", "Right Ventral DC"]
19
20
[NETWORK]
21
num_class = 33
22
num_channels = 1
23
num_filters = 64
24
kernel_h = 5
25
kernel_w = 5
26
kernel_c = 1
27
stride_conv = 1
28
pool = 2
29
stride_pool = 2
30
#Valid options : NONE, CSE, SSE, CSSE
31
se_block = "CSSE"
32
drop_out = 0.2
33
34
[TRAINING]
35
exp_name = "finetuned_alldata_axial"
36
final_model_file = "finetuned_alldata_axial.pth.tar"
37
learning_rate = 1e-4
38
train_batch_size = 5
39
val_batch_size = 5
40
log_nth = 50
41
num_epochs = 10
42
optim_betas = (0.9, 0.999)
43
optim_eps = 1e-8
44
optim_weight_decay = 0.00001
45
lr_scheduler_step_size = 3
46
lr_scheduler_gamma = 0.1
47
48
#Uses the last checkpoint file from the exp_dir_name folder
49
use_last_checkpoint = True
50
use_pre_trained = False
51
pre_trained_path = "saved_models/IXI_fsNet_axial.pth.tar"
52
53
[EVAL]
54
eval_model_path = "saved_models/quicknat_finetuned_1.pth.tar"
55
data_dir = "/home/masterthesis/shayan/nas_drive/Data_Neuro/OASISchallenge/FS"
56
label_dir = "/home/masterthesis/shayan/nas_drive/Data_Neuro/OASISchallenge"
57
volumes_txt_file = "datasets/MALC/test_volumes.txt"
58
#Valid options : Neo, FS
59
remap_config = "Neo"
60
#Valid options : COR, AXI, SAG
61
orientation = "COR"
62
save_predictions_dir = "predictions"
63
data_id = "MALC"