Skip to content

Image Segmentation of Brain Tumors using Convolutional Neural Networks. Implemented U-Net and LinkNet architectures.

Notifications You must be signed in to change notification settings

mhagglun/Image-Segmentation-of-Brain-Tumors

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

56 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Image Segmentation of Brain Tumors using Convolutional Neural Networks

In this project we implement two different networks in order to create segmentation masks for identifying brain tumors.

The complete dataset can be accessed here

Overview

Dataset Used

This brain tumor dataset contains 3064 T1-weighted contrast-inhanced images from 233 patients with three kinds of brain tumor: meningioma (708 slices), glioma (1426 slices), and pituitary tumor (930 slices). Each image has the dimension (512 x 512 x 1). The full dataset is available here

Original Image Original Image With Mask

Model Architecture

The model architecture for U-Net

The model architecture for LinkNet

Training Process

The networks were trained on a system with an AMD Ryzen 5 2600 processor overclocked to 4.0GHz, 16GB of RAM and an NVIDIA GeForce RTX 2070 GPU.

U-Net

We experimented with the number of initial filters and settled for 16. U-net was found to perform well when using a batch size of 8. The initial learning rate was set to 10e−4 .

LinkNet

We experimented with the number of start filters and found that 64 seemed to give the best results. We noticed that LinkNet seemed to be sensitive to parameter settings. LinkNet was trained using an initial learning rate of 5 · 10e−4 and a batch size of 10. Batch normalisation was used after each convolutional layer. We also tried adding L2-regularisation but that did not seem to significantly improve training.

Results

The final DICE scores achieved by the respective networks on the test set were

Method DICE Score
U-Net 0.703
LinkNet 0.715

Some samples from the predictions are shown below

U-Net Linknet

About

Image Segmentation of Brain Tumors using Convolutional Neural Networks. Implemented U-Net and LinkNet architectures.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages