|
import sys |
|
|
|
cnet_models = ["canny", "normal", "depth", "openpose", "hed", "scribble", "mlsd", "seg"] |
|
unet_models = ["sd15", "openjourney", "Nitro-Diffusion", "anything-v3.0", "vintedois-diffusion-v0-1"] |
|
|
|
print(f"# ControlNet Generation Comparison") |
|
print("") |
|
|
|
def print_header(): |
|
print("|seed|", end="") |
|
for ot in unet_models: |
|
print(f"{ot}|", end="") |
|
print("") |
|
|
|
print("|", end="") |
|
for _ in range(len(unet_models) + 1): |
|
print("---|", end="") |
|
print("") |
|
|
|
def get_url(path): |
|
|
|
url = 'https://huggingface.co/takuma104/controlnet_dev/resolve/main/model_compare_full/all_controlnet_models/' |
|
return url + path |
|
|
|
for cnet in cnet_models: |
|
print(f"## ControlNet {cnet}") |
|
print("") |
|
print_header() |
|
for seed in range(16): |
|
print(f"|{seed}|", end="") |
|
for unet in unet_models: |
|
url = get_url(path=f"{cnet}/{unet}/output_{seed:02d}.png") |
|
print(f'<a href="{url}"><img width="128" src="{url}"/></a>|', end="") |
|
print("") |
|
|
