Diffusion Model and ControlNet Based Image

Install

git clone git@github.com:jingzhang00/colorization.git
pip install -r requirements.txt

How to use

python main.py

Then you will get two URLs, one for local and one for the public, open it in the browser then the webpage should look like the following:

NOTE: For negative prompt, we use the default setting "monochrome".

Outcomes

an overview

compare with CNN/GAN based method

results when using different prompts