-
Notifications
You must be signed in to change notification settings - Fork 67
/
config_docker.json
61 lines (61 loc) · 1.73 KB
/
config_docker.json
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
{
"ProjectName": "VNet Tensorflow",
"ProjectDetail": {
"BodyPart": "Liver",
"Diseases": "Lesion"
},
"TrainingSetting": {
"Data": {
"TrainingDataDirectory":"./data/training",
"TestingDataDirectory": "./data/testing",
"ImageFilenames": ["image.nii.gz"],
"LabelFilename": "label.nii.gz"
},
"Restore": true,
"SegmentationClasses": [0,1,2],
"LogDir": "./tmp/log",
"CheckpointDir": "./tmp/ckpt",
"BatchSize": 32,
"PatchShape": [256,256,32],
"ImageLog": true,
"Testing": true,
"Epoches": 99999,
"LogInterval": 25,
"Networks": {
"Name":"VNet",
"Dropout": 0.1
},
"Loss": {
"Name": "weighted_sorensen",
"Weights": [0.01,0.1,1],
"Alpha": 10
},
"Optimizer":{
"Name": "Adam",
"InitialLearningRate": 1e-2,
"Momentum":0.9,
"Decay":{
"Factor": 0.99,
"Steps": 100
}
},
"Spacing": [0.75,0.75,0.75],
"DropRatio": 0.01,
"MinPixel":30
},
"EvaluationSetting":{
"Data":{
"EvaluateDataDirectory": "./data_3DRA/evaluate",
"ImageFilenames": ["CBCT_reg.nii"],
"LabelFilename": "label_tf.nii.gz",
"ProbabilityFilename": "probability_tf.nii.gz"
},
"CheckpointPath": "./tmp/ckpt/checkpoint-23125",
"Stride": [256,256,32],
"BatchSize": 10,
"Pipeline": "./pipeline/pipeline2D.yaml",
"ProbabilityOutput":true,
"LargestConnectedComponent": true,
"VolumeThreshold": 50
}
}