Convolutional Autoencoder for Brain Tumor Segmentation

The notebook applies segementation to the brain tumor dataset.

Training for 30 epoches is for demonstration; some settings, such as the GPU usage, may not have been optimal.

Datset

Brain Tumor

Tools

Google Colab

Google Drive

PyTorch

CUDA

NumPy

Matplotlib

Models Genesis

References

U-Net: Convolutional Networks for Biomedical Image Segmentation

PyTorch Image Segmentation Tutorial for Beginners — I

PyTorch Image Segmentation Tutorial for Beginners — II

How to Evaluate Image Segmentation Models?

Answer to How to Check if Pytorch is Using the GPU?