BE
berkegokmen1/counterfactual-chexray-disease-editing
Official Implementation for the paper titled: "Counterfactual Disease Removal and Generation in Chest X-Rays Using Diffusion Models"
Counterfactual Disease Removal and Generation in Chest X-Rays Using Diffusion Models
Paper | Project Website | BibTeX
Authors
Ahmet Berke Gökmen, Ender Konukoglu
TODO
- Release Website
- Release Code
- Run Instructions
Setup
conda create -n chexray-editing python=3.10
pip install -r requirements.txt [TODO]Inference
Please download chexzero, chexpert and chexray-diffusion checkpoints from their respective repositories and update the paths in config.yaml.
In additon to the checkpoints, you'll need to download CheXpert-v1.0-small dataset from the official chexpert website or you may use any chest x-ray image.
python finetune_sample.py --config config.yaml --target "Pleural Effusion" --mode "removal" --experiment_name "demo"Questions
You may reach me through LinkedIn.
This work would not have been possible without:
- https://github.com/rajpurkarlab/CheXzero
- https://github.com/jfhealthcare/Chexpert
- https://github.com/saiboxx/chexray-diffusion
BibTeX