cxr_gen_models
Collection
7 items β’ Updated
This model is a fine-tuned version of Stable Diffusion v1.5 trained to generate synthetic chest X-ray (CXR) images.
It is designed to produce medically relevant radiographs based on textual prompts such as disease conditions.
import torch
from diffusers import StableDiffusionPipeline
pipe = StableDiffusionPipeline.from_pretrained(
"P-RAJIV/cxr_stable_diffusion_v1_5",
torch_dtype=torch.float16
).to("cuda")
prompt = "Chest X-ray showing pneumonia"
image = pipe(prompt).images[0]
image.save("output.png")
π§ͺ Example Prompts
"Normal chest X-ray"
"Chest X-ray showing cardiomegaly"
"Lung opacity in right lower lobe"
"Severe pneumonia chest radiograph"
β οΈ Limitations
Generated images are synthetic and not for clinical use
May produce anatomically inconsistent outputs
Performance depends heavily on prompt quality
π Training Details
Base Model: Stable Diffusion v1.5
Domain: Chest X-ray imaging
Fine-tuning: Text-to-image diffusion training
π§ Intended Use
Research in medical imaging
Data augmentation
Diffusion model experimentation
β Disclaimer
This model is not intended for medical diagnosis or clinical decision-making.