/Brain-tumor-segmentation

Mini-project using the TensorFlow and PyTorch framework jointly

Primary LanguagePython

Brain-tumor-segmentation

Segmenting brain tumor by using various networks from these papers:

  1. U-net: Convolutional networks for biomedical image segmentation
  2. ResUNet-a: A deep learning framework for semantic segmentation of remotely sensed data
  3. Resunet++: An advanced architecture for medical image segmentation

Project Topic & Goal

The main goal of this project is the brain tumor segmentation.

  • As the use of deep learning in the medical field is gradually increasing, high diagnostic performance is being derived.
  • Furthermore, in order to utilize medical data, the need to automate data analysis and purification, which was previously handled by experts, is gradually increasing.
  • If a high-accuracy and generalizable model can be used, the time that radiologists spend on data analysis can be effectively reduced.

Dataset

We downloaded the dataset from Kaggle.

  1. Data format: .tif
  2. Total: 110 dataset (Train/Validation-100, Test-10)
  3. Ground-truth data: Mask data

Method & Performance

In order to select a model that can improve performance, we compared three models with promising performance in the segmentation task. brain_net

Qualitative evaluation (segmented mask visualization)

brain_res

Quantitative evaluation

brain_1