U-Net Brain Tumor Segmentation
This repo show you how to train a U-Net for brain tumor segmentation. By default, you need to download the training set of BRATS 2017 dataset, which have 210 HGG and 75 LGG volumes, and put the data folder along with all scripts.
data
-- Brats17TrainingData
-- train_dev_all
model.py
train.py
...
About the data
Note that according to the license, user have to apply the dataset from BRAST, please do NOT contact me for the dataset. Many thanks.
Fig 1: Brain Image
- Each volume have 4 scanning images: FLAIR、T1、T1c and T2.
- Each volume have 4 segmentation labels:
Label 0: background
Label 1: necrotic and non-enhancing tumor
Label 2: edema
Label 4: enhancing tumor
The prepare_data_with_valid.py split the training set into 2 folds for training and validating. By default, it will use only half of the data for the sake of training speed, if you want to use all data, just change DATA_SIZE = 'half' to all.
About the method
-
Network and Loss: In this experiment, as we use dice loss to train a network, one network only predict one labels (Label 1,2 or 4). We evaluate the performance using hard dice and IOU.
-
Data augmenation: Includes random left and right flip, rotation, shifting, shearing, zooming and the most important one -- Elastic trasnformation, see "Automatic Brain Tumor Detection and Segmentation Using U-Net Based Fully Convolutional Networks" for details.
Fig 2: Data augmentation
Start training
We train HGG and LGG together, as one network only have one task, set the task to all, necrotic, edema or enhance, "all" means learn to segment all tumors.
python train.py --task=all
Note that, if the loss stick on 1 at the beginning, it means the network doesn't converge to near-perfect accuracy, please try restart it.
Citation
If you find this project useful, we would be grateful if you cite the TensorLayer paper：
@article{tensorlayer2017,
author = {Dong, Hao and Supratak, Akara and Mai, Luo and Liu, Fangde and Oehmichen, Axel and Yu, Simiao and Guo, Yike},
journal = {ACM Multimedia},
title = {{TensorLayer: A Versatile Library for Efficient Deep Learning Development}},
url = {http://tensorlayer.org},
year = {2017}
}
cent call last):
File "train.py", line 250, in 
39 Jan 04, 2023
9 Mar 03, 2022
594 Jan 05, 2023
32 Dec 06, 2022
62 Dec 21, 2022
845 Jan 03, 2023
169 Dec 02, 2022
21 Jan 05, 2022
83 Jan 05, 2023
40 Dec 05, 2022
12 Nov 05, 2022
4.4k Dec 25, 2022
9 Nov 14, 2022
191 Dec 16, 2022
24 Oct 15, 2022
152 Dec 22, 2022
4 Aug 28, 2022
77 Dec 16, 2022
336 Dec 30, 2022
87 Dec 21, 2022