File size: 473 Bytes
5954ed4 fd1011e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
---
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)
|