settings.ini 2.09 KB
Newer Older
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
1
[DATA]
2
data_folder_name = "datasets"
3
use_data_file = False
4
5
data_directory = "/well/win-biobank/projects/imaging/data/data3/subjectsAll/"
data_file = "/well/win-biobank/projects/imaging/data/data3/subjectsAll/subj_22k.txt"
6
k_fold = None
7
data_split_flag = False
8
test_percentage = 5
9
subject_number = 600
10
11
12
train_list = "datasets/train.txt"
validation_list = "datasets/validation.txt"
test_list = "datasets/test.txt"
13
scaling_factors = "datasets/scaling_factors.pkl"
14
15
16
17
train_data_file = "dMRI/autoptx_preproc/tractsNormSummed.nii.gz"
train_output_targets = "fMRI/rfMRI_25.dr/dr_stage2.nii.gz"
validation_data_file = "dMRI/autoptx_preproc/tractsNormSummed.nii.gz"
validation_target_file = "fMRI/rfMRI_25.dr/dr_stage2.nii.gz"
18
brain_mask_path = "utils/MNI152_T1_2mm_brain_mask.nii.gz"
19
mean_mask_path = "utils/mean_dr_stage2.nii.gz"
20
mean_reduction = True
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
21
22

[TRAINING]
23
24
25
26
27
experiment_name = "CU3D17-3"
pre_trained_path = "saved_models/CU3D17-3.pth.tar"
final_model_output_file = "CU3D17-3.pth.tar"
training_batch_size = 3
validation_batch_size = 3
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
28
use_pre_trained = False
29
learning_rate = 1e-1
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
30
31
32
optimizer_beta = (0.9, 0.999)
optimizer_epsilon = 1e-8
optimizer_weigth_decay = 1e-5
33
number_of_epochs = 200
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
34
loss_log_period = 50
35
learning_rate_scheduler_step_size = 5
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
36
learning_rate_scheduler_gamma = 1e-1
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
37
use_last_checkpoint = False
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
38
39

[NETWORK]
40
41
42
kernel_heigth = 3
kernel_width = 3
kernel_depth = 3
43
; kernel_classification = 1
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
44
45
46
47
input_channels = 1
output_channels = 64
convolution_stride = 1
dropout = 0.2
48
; pool_kernel_size = 2
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
49
50
51
pool_stride = 2
up_mode = "upconv"
number_of_classes = 1
52
53
54
55
; ---> parameters for the ResNet CGAN
pool_kernel_size = 3
kernel_classification = 7

Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
56
57
58
59

[MISC]
save_model_directory = "saved_models"
logs_directory = "logs"
60
checkpoint_directory = "checkpoints"
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
61
experiments_directory = "experiments"
62
device = 0
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
63

64
[EVALUATION]
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
65
66
67
68
69
70
71
trained_model_path = "saved_models/model.pth.tar"
data_directory = "/well/win-biobank/projects/imaging/data/data3/subjectsAll/"
data_file = "/dMRI/autoptx_preproc/tractsNormSummed.nii.gz"
output_targets = "/fMRI/rfMRI_25.dr/dr_stage2.nii.gz"
data_list = "test.txt"
orientation = "coronal"
saved_predictions_directory = "predictions"
72