---
license: mit
tags:
  - image-segmentation
  - medical-imaging
  - pytorch
  - unet
datasets:
  - custom
inference: true
---

# 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)