|
import sys |
|
|
|
cnet_models = ["canny", "normal", "depth", "openpose", "hed", "scribble", "mlsd", "seg"] |
|
processors = ["ref", "diffusers"] |
|
image_types = ['bird', 'human', 'room', 'vermeer'] |
|
|
|
print(f"# ControlNet Generation Comparison") |
|
print("") |
|
|
|
def print_header(): |
|
print("|seed|", end="") |
|
for proc in processors: |
|
print(f"{proc} with prompt|", end="") |
|
for proc in processors: |
|
print(f"{proc} guess_mode|", end="") |
|
for proc in ['diffusers']: |
|
print(f"{proc} guess_mode with prompt|", end="") |
|
print("") |
|
|
|
print("|", end="") |
|
for _ in range(len(processors) * 2 + 1 + 1): |
|
print("---|", end="") |
|
print("") |
|
|
|
def get_url(path): |
|
|
|
url = 'https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/' |
|
return url + path |
|
|
|
for cnet in cnet_models: |
|
print(f"## ControlNet {cnet}") |
|
print("") |
|
print_header() |
|
|
|
for image_type in image_types: |
|
for seed in range(4): |
|
if cnet=='openpose' and image_type=='human' and seed == 3: |
|
continue |
|
|
|
print(f"|{seed}|", end="") |
|
for proc in processors: |
|
url = get_url(path=f"{proc}/output_{image_type}_{cnet}_{seed:01d}.png") |
|
print(f'<a href="{url}"><img width="128" src="{url}"/></a>|', end="") |
|
|
|
for proc in processors: |
|
url = get_url(path=f"{proc}/output_{image_type}_{cnet}_{seed:01d}_gm.png") |
|
print(f'<a href="{url}"><img width="128" src="{url}"/></a>|', end="") |
|
|
|
for proc in ['diffusers']: |
|
url = get_url(path=f"{proc}/output_{image_type}_{cnet}_{seed:01d}_gm_wp.png") |
|
print(f'<a href="{url}"><img width="128" src="{url}"/></a>|', end="") |
|
print("") |
|
|
