BC-DPM

This repository is heavily based on improved diffusion and guided diffusion.

Usage

Step1: Train a diffusion model on normal dose images by running scripts/image_train.py.

Step2: Sample high dose images from low dose images by running scripts/bc_dpm_sample.py.

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
.idea		.idea
guided_diffusion		guided_diffusion
ref_imgs/init0.001_tiny		ref_imgs/init0.001_tiny
scripts		scripts
README.md		README.md
resizer.py		resizer.py