teohyc/Covid-XRay-Diffusion-Model

Diffusion Model for COVID-19 X-ray Generation

This is a diffusion model designed for generating synthetic COVID-19 X-ray images. The model takes random noise as input and iteratively denoises it to produce realistic X-ray images. Used to generate synthetic xray image for scarce COVID-19 positive cases, which can be used for data augmentation in training diagnostic models.

Training data from https://data.mendeley.com/datasets/9xkhgts2s6/4 Full project file at https://github.com/teohyc/covidxraydiffusion

##Usage

from diffusers import DDPMPipeline
import matplotlib.pyplot as plt

# Load the pipeline
pipeline = DDPMPipeline.from_pretrained("teohyc/Covid-XRay-Diffusion-Model")

# Generate a synthetic X-ray
image = pipeline(num_inference_steps=500).images[0] #default is 1000 steps, but you can reduce it for faster generation (at the cost of quality)

# Display
plt.imshow(image, cmap='gray')
plt.axis('off')
plt.show()