metadata

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)