Skip to main content
. 2020 Oct 28;69(6):443–449. doi: 10.2144/btn-2020-0108

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