U-Net: Semantic segmentation with PyTorch https://arxiv.org/abs/1505.04597

Install dependencies

pip install -r requirements.txt

Run training:

python3 train.py

Testing

python3 test.py

Data

Link: https://www.kaggle.com/datasets/nikhilroxtomar/brain-tumor-segmentation