This repository demonstrates the utilization of UNETR for brain tumor segmentation.
 |
|---|
| The block diagram of the Original UNETR model. |
The dataset contains 3064 pairs of MRI brain images and their respective binary mask indicating tumor.
Download the dataset: Brain Tumor Segmentation
Original Dataset: Brain Tumor Segmentation
The sequence in the images below is Input Image, Ground Truth and Prediction.
 |
|---|
 |
 |
 |
 |
- Train on more epochs.
- Increase the input image resolution.
- Apply data augmentation.
For more follow me on: