1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
{
"mode": "training",
"common":{
"base_path": "/hdd/disk4/Segmentation/Weights",
"decoder": "NoduleSegDecoder_proxima",
"encoder": "NoduleSegEncoder_proxima",
"initial_learning_rate": 0.001,
"learning_rate_drop": 0.65,
"learning_rate_patience": 2,
"base_lr":0.0001,
"max_lr":1e-3,
"step_size_up":4000,
"model_parameters": "modelParams/model_parameters_nodules.json",
"normalizer": [
-1024.0,
400.0
],
"train_file": "dataFiles/NoDensity/train_files.json",
"val_file": "dataFiles/NoDensity/val_files.json",
"test_file": "dataFiles/test_files.json",
"stride_ratio" :8,
"min_patch_size":12,
"diameter_enlarge_ratio":3,
"diameter_idx":4,
"spacing_idx":8,
"densityTypeIndex":-4,
"direction_z_ratio":0.5,
"T_loss_alpha":0.6,
"do_cls":false,
"surface_loss":true,
"e_value":0.1,
"compare_loss_ratio":0,
"zoom_shape":64,
"calculate_shape_choice":2
},
"training": {
"aug": true,
"n_epochs": 30,
"num_slice": -1,
"pre_model_file": "",
"model_name":"NoduleSegUnet",
"model_order":2,
"prefix": "BasicLRNewCrop",
"train_batch_size": 4,
"val_batch_size": 4,
"voxel_size": -1,
"nonfix_crop":true,
"save_cp":true,
"repeat_times":3,
"compare_choice":false
},
"test":
{
"num_slice": -1,
"num_units": 1,
"pre_model_file": "/hdd/disk4/Segmentation/Weights/0114_NS_lr0.001_NewData_e0.1_NoCompare64_Alpha0.6/01_NewModelParameters/CP_epoch23_loss_0.05.pth",
"voxel_size": -1,
"nonfix_crop":true,
"infer_idx":-1,
"save_result_choice":false,
"badcase_vis":false,
"diameter_ths":"[0,100000]",
"dice_range":"[0,0.5]"
}
}