medical-image-generation

Fine tuning control net for medical chest x ray images

Training

python training.py

This file adds control net archirtecture to stable diffusion model

python tool_add_control_sd21.py ./models/v2-1_512-ema-pruned.ckpt ./models/control_sd21_ini.ckpt

python filter.py

Name		Name	Last commit message	Last commit date
Latest commit History 31 Commits
ControlNet		ControlNet
README.md		README.md
convert_to_jpg.py		convert_to_jpg.py
data_curation.ipynb		data_curation.ipynb
filter.py		filter.py
training.ipynb		training.ipynb
training.py		training.py