This repository contains a TensorFlow implementation of Semantic Segmentation of Brain Tumor on multi-band 3D volumes using non-uniform 3D UNets
This codebase provides:
- Testing code
- Visualization code
- Python 3
- TensorFlow tested on 2.4.1
- os
- numpy
- tqdm
- Keras
- glob
- Segmentation Models 3D
Run the following code to install all pip packages:
python3 ./scripts/pkg_install.py- run the following script to download the generated and conerted data (in pkl format) and copy them under
the root folder ./
sh ./scripts/download_generated_pkl_files.sh - run the following script to download the full dataset
sh ./scripts/download_full_data.sh The model definations are shown here:
Figure: Detailed parameters for all the five models
w/ all the model's performance:
Figure: Info. and statistical data of the models