This research is about segmentation for throatic and lumbar spine using deep learning techniques. Convolutional neural network has been employed. Furthermore, there has been UNet-2D and Unet-3D applied as its type of architecture.
As per the augmentation techniques, gaussian, rescaling, blur, rotation applied to improve the generalization aspect of the model and also diceCoeficient and diceLoss for the evaluation metrics.

Training with this data might take up to few days to finish depending on the batch size and hardware limitations. In a Meanwhile the validation error between Unet-2D and Unet-3D has been compared.

Unet 3d is just another version of unet2d where it uses 3D filters in its computation of convolution. All trained models will be validated on a never seen data called the test data and later will be compared with each other.

Here you can see the model architecture for both 2d and 3d. Depending on the model architecture, UNet 3d or UNet 2d, the input data must be designed accordingly.

The necessary packages that needs to be installed can be found in requirement.txt file.

• numpy
• future
• gast
• tensorflow
• tensorflow-addons
• pynrrd
• pickle-mixin
• scipy
• h5py
• pillow
• volumentations-3D
• keras
• pydot

1. Create a text file containing the list of nrrd image files and their segmentation seperated by comma and line. As an example:
   (path-to-first-nrrd-image-file),(path-to-first-nrrd-label-file)
(path-to-second-nrrd-image-file),(path-to-second-nrrd-label-file)
   . . .

2. Use the adreess to the text file and launch makeCube.py to create a 3D matrix that contains all the images together with their augmented versions.
   

3. Use checkCube.py to observe and make sure that the matrix is created correctly.
   

4. Use train.py to train the model and predict the labels.
   

The data is taken from different sources:

The result for the comparison of Dice Score for Unet-2D and Unet-3D is presented in Table below. The result explains the model validation trained on different combination of training datasets.

Training accuracy over epochs for Unet-2D and Unet-3D of one of the combination datasets has shown in follow.