Covid XRay Diffusion Model

##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()

--- tags: - Lung - Pneumonia - Covid-19 - PyTorch license: mit language: - en pipeline_tag: unconditional-image-generation library_name: diffusers --- --- # 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/covid_xray_diffusion ##Usage ```python 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() ```