|
a |
|
b/resources/config.ini |
|
|
1 |
[basics] |
|
|
2 |
train_flag: false |
|
|
3 |
compute_patches: false |
|
|
4 |
resume: false |
|
|
5 |
|
|
|
6 |
test_flag: true |
|
|
7 |
uncertainty_flag: true |
|
|
8 |
|
|
|
9 |
plot: false |
|
|
10 |
tensorboard_logs: tensorboard_logs/ |
|
|
11 |
|
|
|
12 |
[uncertainty] |
|
|
13 |
n_iterations: 20 |
|
|
14 |
# uncertainty_type: ttd |
|
|
15 |
uncertainty_type: tta |
|
|
16 |
use_dropout: false |
|
|
17 |
|
|
|
18 |
[model] |
|
|
19 |
model_path_local: /Users/lauramora/Documents/MASTER/TFM/Code/BrainTumorSegmentation/results/checkpoints/last_models/ |
|
|
20 |
model_path_server: /mnt/gpid07/users/laura.mora/results/checkpoints/ |
|
|
21 |
|
|
|
22 |
# newer checks! |
|
|
23 |
# checkpoint: last_models/checkpoint_epoch_349_val_loss_0.33079595190204986_dice_0.6692040464649461.pth |
|
|
24 |
# checkpoint: model_1598640005/checkpoint_epoch_168_val_loss_0.20105469390137554_dice_0.7989453060986245.pth |
|
|
25 |
# checkpoint: model_1598639885/checkpoint_epoch_198_val_loss_0.19342842820572526_dice_0.8065715717942747.pth |
|
|
26 |
checkpoint: model_1598640035/checkpoint_epoch_142_val_loss_0.21437616135976087_dice_0.7856238380039416.pth |
|
|
27 |
# checkpoint: model_1598550861/checkpoint_epoch_215_val_loss_0.2378825504485875_dice_0.7621174487349105.pth |
|
|
28 |
# checkpoint: model_1598651693/checkpoint_epoch_297_val_loss_0.3553243059001557_dice_0.644675692108026.pth |
|
|
29 |
# checkpoint: model_1598651693/checkpoint_epoch_316_val_loss_0.33101850666411936_dice_0.6689814933358806.pth |
|
|
30 |
|
|
|
31 |
|
|
|
32 |
loss: gdl |
|
|
33 |
eval_regions: false |
|
|
34 |
init_features_maps: 32 |
|
|
35 |
|
|
|
36 |
n_epochs: 100 |
|
|
37 |
# network: 3dunet |
|
|
38 |
network: 3dunet_residual |
|
|
39 |
# network: vnet_asymm |
|
|
40 |
|
|
|
41 |
# unet |
|
|
42 |
unet_order: crg |
|
|
43 |
# cli - conv + LeakyReLU + instancenorm |
|
|
44 |
|
|
|
45 |
# vnet asymm |
|
|
46 |
non_linearity: relu |
|
|
47 |
kernel_size: 3 |
|
|
48 |
padding: 1 |
|
|
49 |
# vnet |
|
|
50 |
use_elu: true |
|
|
51 |
|
|
|
52 |
|
|
|
53 |
# optimizer |
|
|
54 |
optimizer: ADAM |
|
|
55 |
learning_rate: 1e-4 |
|
|
56 |
weight_decay: 1e-5 |
|
|
57 |
|
|
|
58 |
## sgd only |
|
|
59 |
momentum: 0.99 |
|
|
60 |
|
|
|
61 |
# scheduler LR |
|
|
62 |
scheduler: true |
|
|
63 |
patience: 30 |
|
|
64 |
## by a factor of 5 |
|
|
65 |
lr_decay: 0.2 |
|
|
66 |
|
|
|
67 |
[dataset] |
|
|
68 |
dataset_root_path_server: /mnt/gpid07/users/laura.mora/datasets/2020/ |
|
|
69 |
dataset_root_path_local: /Users/lauramora/Documents/MASTER/TFM/Data/2020/ |
|
|
70 |
|
|
|
71 |
dataset_train_folder: train |
|
|
72 |
dataset_val_folder: validation |
|
|
73 |
dataset_test_folder: test |
|
|
74 |
|
|
|
75 |
train_csv: brats20_data.csv |
|
|
76 |
test_csv: brats20_data.csv |
|
|
77 |
val_csv: brats20_val.csv |
|
|
78 |
|
|
|
79 |
classes: 4 |
|
|
80 |
n_modalities: 4 |
|
|
81 |
# Use dataloader |
|
|
82 |
batch_size: 2 |
|
|
83 |
lgg_only: false |
|
|
84 |
|
|
|
85 |
# If using sampler |
|
|
86 |
n_patients_per_batch: 8 |
|
|
87 |
n_patches: 1 |
|
|
88 |
# source_sampling: src.dataset.patching.centered_crop_patch |
|
|
89 |
# source_sampling: src.dataset.patching.random_tumor_distribution |
|
|
90 |
source_sampling: src.dataset.patching.no_patch |
|
|
91 |
# sampling_method: src.dataset.patching.no_patch |
|
|
92 |
sampling_method: src.dataset.patching.random_tumor_distribution |
|
|
93 |
# sampling_method: src.dataset.patching.random_distribution |
|
|
94 |
# sampling_method: src.dataset.patching.binary_distribution |
|
|
95 |
patch_size: 64 |
|
|
96 |
64 |
|
|
97 |
64 |