DeepLearningSpineMRISegmentation/default_config.json at main · mytkom/DeepLearningSpineMRISegmentation · GitHub

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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
{
    "training": {
        "saveDir": "models",
        "visNum": 5,
        "earlyStoppingPatience": 3,
        "earlyStoppingDelta": 0.001,
        "learningRate": 0.0001,
        "weightDecay": 0.001,
        "numEpochs": 15,
        "device": "cpu",
        "seed": null,
        "schedulerType": "ReduceLROnPlateau",
        "schedulerWarmupEpochs": 0,
        "schedulerParams": {
            "patience": 3,
            "factor": 0.5,
            "min_lr": 1e-06,
            "mode": "min"
        },
        "lossFnType": "DiceFocalLoss",
        "lossFnParams": {
            "softmax": true,
            "to_onehot_y": true,
            "lambda_dice": 2.0,
            "lambda_focal": 1.0
        }
    },
    "preprocessing": {
        "spiderPath": "data/SPIDER",
        "spiderPrefix": "SPIDER",
        "spiderImagesPath": "data/SPIDER/images",
        "spiderMasksPath": "data/SPIDER/masks",
        "osfPath": "data/osf-files",
        "osfPrefix": "OSF",
        "spineOutputPath": "data/spine_output",
        "spineOutputPrefix": "Spine_Output",
        "targetSpacing": [
            1.0,
            1.0,
            1.0
        ],
        "minPixelsMask": 50
    },
    "dataloader": {
        "batchSize": 8,
        "numWorkers": 2,
        "valSplit": 0.2,
        "testSplit": 0.1,
        "distorsionProb": 0.3,
        "distorsionNumCells": 5,
        "distorsionDistortLimit": 0.03,
        "noiseProb": 0.1,
        "noiseMean": 0.0,
        "noiseStd": 0.1,
        "contrastProb": 0.2,
        "contrastGamma": [
            0.5,
            2.0
        ],
        "rotationProb": 0.5,
        "flipProb": 0.5,
        "cropSize": [
            256,
            256
        ],
        "cropPos": 2.0,
        "cropNeg": 1.0,
        "cropNumSamples": 1,
        "swOverlap": 0.5
    },
    "model": {
        "modelType": "tri_conv_unext",
        "inChannels": 1,
        "compile": true,
        "classes": 4,
        "baseC": 32,
        "bilinear": true,
        "triConvUnext": {
            "useDeformable": true,
            "useDilated": true,
            "useDepthwise": true
        }
    },
    "wandb": {
        "project": "spine-segmentation"
    },
    "preprocessedDataDir": "data/preprocessed_v3",
    "cacheDir": "data/cache",
    "runIdentifier": "default"
}