Setup

prepare.py: Data preprocessing. Here we resize all Flickr-8k images to (256, 256) and pixelate them.

train_controlnet.py: Train ControlNet on pixelated Flickr-8k with denoising condition.

inference.py: Evaluate ControlNet.

Demonstration

See 6_8301_final_project_eval.ipynb

Name		Name	Last commit message	Last commit date
Latest commit History 11 Commits
6_8301_final_project.ipynb		6_8301_final_project.ipynb
6_8301_final_project_eval.ipynb		6_8301_final_project_eval.ipynb
README.md		README.md
inference.py		inference.py
prepare.py		prepare.py
train.sh		train.sh
train_controlnet.py		train_controlnet.py