ge-xing/Diff-UNet
Diff-UNet: A Diffusion Embedded Network for Volumetric Segmentation. (using diffusion for 3D medical image segmentation)
Embeds diffusion model components directly within a UNet encoder-decoder architecture to progressively refine segmentation predictions across volumetric medical images. Demonstrates superior performance on multi-class segmentation tasks including brain tumors (BraTS2020) and multi-organ segmentation (BTCV), supporting datasets with varying modality counts and target classes. Built with PyTorch and includes end-to-end training/testing pipelines for 3D volumetric data.
192 stars. No commits in the last 6 months.
Stars
192
Forks
29
Language
Python
License
Apache-2.0
Category
Last pushed
Mar 22, 2024
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/diffusion/ge-xing/Diff-UNet"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related models
LemuelPuglisi/BrLP
[MICCAI, 2024] (Oral, RU @ MedIA Best Paper Award) Official implementation of the Brain Latent...
GabrieleLozupone/LDAE
Official PyTorch implementation of "Latent Diffusion Autoencoders: Toward Efficient and...
Warvito/generative_chestxray
Repository to train Latent Diffusion Models on Chest X-ray data (MIMIC-CXR) using MONAI...
jwmao1/MedSegFactory
[ICCV 2025] MedSegFactory: Text-Guided Generation of Medical Image-Mask Pairs
sandialabs/pvcracks
Si-PV cell crack image recognition, modeling, and power loss prediction