DiffuAug

This repository contains code for training a model to generate breast MRI using Classifier-Free-Diffusion-Guidance and sampling with DDPM and DDIM.

Generation Results

With tumors

Without tumors

Verifying Augmentation Performance through Generative Models

Setup

conda

conda env create -f DiffuAug
conda activate DiffuAug

pip

pip install -r requirements.txt

Acknowledgement

This work was supported by Institute of Information & communications Technology Planning & Evaluation (IITP) grant funded by the Korea government(MSIT) (No.1711195685, Development of object-oriented synthetic data generation and evaluation methods)