forked from jackyko1991/vnet-tensorflow
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config_sample.json
56 lines (56 loc) · 1.54 KB
/
config_sample.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
{
"ProjectName": "VNet Tensorflow",
"ProjectDetail": {
"BodyPart": "Liver",
"Diseases": "Lesion"
},
"TrainingSetting": {
"Data": {
"TrainingDataDirectory":"./data/training",
"TestingDataDirectory": "./data/testing",
"ImageFilenames": ["image.nii"],
"LabelFilename": "label.nii"
},
"Restore": true,
"SegmentationClasses": [0,1,2],
"LogDir": "./tmp/log",
"CheckpointDir": "./tmp/ckpt",
"BatchSize": 32,
"PatchShape": [256,256],
"ImageLog": false,
"Testing": true,
"TestStep": 30,
"Epoches": 99999,
"MaxIterations": 15000,
"LogInterval": 50,
"Networks": {
"Name":"VNet",
"Dropout": 0.01
},
"Loss": "sorensen",
"Optimizer":{
"Name": "Adam",
"InitialLearningRate": 1e-2,
"Momentum":0.9,
"Decay":{
"Factor": 0.99,
"Steps": 100
}
},
"Spacing": [0.75,0.75],
"DropRatio": 0.01,
"MinPixel":30
},
"EvaluationSetting":{
"Data":{
"EvaluateDataDirectory": "./data/evaluate",
"ImageFilenames": ["image.nii"],
"LabelFilename": "label_tf.nii.gz",
"ProbabilityFilename": "probability_tf.nii.gz"
},
"CheckpointPath": "./tmp/ckpt/checkpoint-23125",
"Stride": [256,256],
"BatchSize": 10,
"ProbabilityOutput":true
}
}