This is the official implementation of the paper "BerDiff: Conditional Bernoulli Diffusion Model for Medical Image Segmentation" at the 26th International Conference on Medical Image Computing and Computer-Assisted Intervention (MICCAI 2023). The pre-print version can be found in arxiv; a camera-ready version will be soon released.
Dec 2023: initial commit.
The LIDC-IDRI dataset can be downloaded from data link. It is preprocessed and given by Probabilistic UNet
LIDC/
|--train/
|--images/
|--LIDC-IDRI-0001/
|--z-105.0_c0.png
|--z-107.0_c0.png
...
|--LIDC-IDRI-0001/
...
|--gt/
|--LIDC-IDRI-0001/
|--z-105.0_c0_l0.png
|--z-105.0_c0_l1.png
|--z-105.0_c0_l2.png
|--z-105.0_c0_l3.png
|--z-107.0_c0_l0.png
|--z-107.0_c0_l1.png
|--z-107.0_c0_l2.png
|--z-107.0_c0_l3.png
...
|--LIDC-IDRI-0001/
...
|--val/
|--test/
A checkpoint can be downloaded from [Google Drive](https://drive.google.com/drive/folders/1HxsR6NJOFILcZzLrcXKfIttctlhdl71I?usp=sharing)
- Linux Platform
- torch==1.12.1+cu102 # depends on the CUDA version of your machine
- torchvision==0.13.1+cu102
- Python==3.9.7
- numpy==1.21.2
- blobfile==2.0.0
- argparse==1.4.0
Code copied a lot from openai/improved-diffusion
If you find our work and code helpful, please kindly cite the corresponding paper:
@article{chen2023ascon,
title={BerDiff: Conditional Bernoulli Diffusion Model for Medical Image Segmentation},
author={Chen, Tao and Wang, Chenhui and Shan, Hongming},
journal={MCCAI 2023},
year={2023}
}