Table 1. . U-net convolutional neural network parameters.
Category | Parameter | Brain | Heart | Nerve |
---|---|---|---|---|
Augmentation | Flip horizontally | Yes | Yes | Yes |
Flip vertically | Yes | Yes | Yes | |
Rotation (degrees) | 0–171 | 0–171 | 0–171 | |
Scale (%) | 90–110 | 85–115 | 90–110 | |
Training data | # of slices | 12 | 10 | 7 |
Validation data | Size | 20% of training data | 20% of training data | 20% of training data |
Training | Input patch size | 64 | 64 | 64 |
Stride to input | 0.6 | 0.6 | 0.6 | |
Epochs | 1000 | 2000 | 100 | |
Batch size | 64 | 64 | 64 | |
Loss function | Categorical cross-entropy | Categorical cross-entropy | Categorical cross-entropy | |
Optimization algorithm | Adadelta | Adadelta | Adadelta |