AndaiMD/brain-unet-model

Brain U-Net for MRI Segmentation

This repository contains a custom U-Net model trained to segment brain MRI images into white matter, gray matter, and cerebrospinal fluid (CSF) compartments.

• Format: PyTorch .pth state dict
• Input: Grayscale 256x256 MRI images
• Output: Segmentation map with 3 classes (WM, GM, CSF)