update for params scale=4.0, steps=50
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- gen_compare_guess_mode/README.md +0 -1
- gen_compare_guess_mode/gen_diffusers_image.py +4 -11
- gen_compare_guess_mode/gen_markdown.py +4 -0
- gen_compare_guess_mode/gen_reference_image.py +1 -1
- gen_compare_guess_mode/output_images/diffusers/output_bird_canny_0.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_canny_0_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_canny_1.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_canny_1_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_canny_2.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_canny_2_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_canny_3.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_canny_3_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_depth_0.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_depth_0_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_depth_1.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_depth_1_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_depth_2.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_depth_2_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_depth_3.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_depth_3_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_hed_0.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_hed_0_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_hed_1.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_hed_1_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_hed_2.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_hed_2_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_hed_3.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_hed_3_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_mlsd_0.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_mlsd_0_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_mlsd_1.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_mlsd_1_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_mlsd_2.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_mlsd_2_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_mlsd_3.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_mlsd_3_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_normal_0.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_normal_0_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_normal_1.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_normal_1_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_normal_2.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_normal_2_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_normal_3.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_normal_3_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_openpose_0.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_openpose_0_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_openpose_1.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_openpose_1_gm.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_openpose_2.png +2 -2
- gen_compare_guess_mode/output_images/diffusers/output_bird_openpose_2_gm.png +2 -2
gen_compare_guess_mode/README.md
CHANGED
|
@@ -71,7 +71,6 @@
|
|
| 71 |
|0|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_0.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_0.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_0.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_0.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_0_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_0_gm.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_0_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_0_gm.png"/></a>|
|
| 72 |
|1|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_1.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_1.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_1.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_1.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_1_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_1_gm.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_1_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_1_gm.png"/></a>|
|
| 73 |
|2|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_2.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_2.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_2.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_2.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_2_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_2_gm.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_2_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_2_gm.png"/></a>|
|
| 74 |
-
|3|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_3.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_3.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_3.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_3.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_3_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_3_gm.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_3_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_3_gm.png"/></a>|
|
| 75 |
|0|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_0.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_0.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_0.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_0.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_0_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_0_gm.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_0_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_0_gm.png"/></a>|
|
| 76 |
|1|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_1.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_1.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_1.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_1.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_1_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_1_gm.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_1_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_1_gm.png"/></a>|
|
| 77 |
|2|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_2.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_2.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_2.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_2.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_2_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_2_gm.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_2_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_2_gm.png"/></a>|
|
|
|
|
| 71 |
|0|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_0.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_0.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_0.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_0.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_0_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_0_gm.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_0_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_0_gm.png"/></a>|
|
| 72 |
|1|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_1.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_1.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_1.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_1.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_1_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_1_gm.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_1_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_1_gm.png"/></a>|
|
| 73 |
|2|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_2.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_2.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_2.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_2.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_2_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_human_openpose_2_gm.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_2_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_human_openpose_2_gm.png"/></a>|
|
|
|
|
| 74 |
|0|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_0.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_0.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_0.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_0.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_0_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_0_gm.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_0_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_0_gm.png"/></a>|
|
| 75 |
|1|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_1.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_1.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_1.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_1.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_1_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_1_gm.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_1_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_1_gm.png"/></a>|
|
| 76 |
|2|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_2.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_2.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_2.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_2.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_2_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/ref/output_room_openpose_2_gm.png"/></a>|<a href="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_2_gm.png"><img width="128" src="https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/diffusers/output_room_openpose_2_gm.png"/></a>|
|
gen_compare_guess_mode/gen_diffusers_image.py
CHANGED
|
@@ -7,7 +7,7 @@ import torch
|
|
| 7 |
import sys
|
| 8 |
import os
|
| 9 |
|
| 10 |
-
from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
|
| 11 |
|
| 12 |
from PIL import Image
|
| 13 |
|
|
@@ -19,10 +19,9 @@ def generate_image(seed, prompt, negative_prompt, control, guess_mode=False):
|
|
| 19 |
image = pipe(
|
| 20 |
prompt=prompt,
|
| 21 |
negative_prompt=negative_prompt,
|
| 22 |
-
guidance_scale=9.0,
|
| 23 |
-
num_inference_steps=20,
|
| 24 |
latents=latent,
|
| 25 |
-
#generator=torch.Generator(device="cuda").manual_seed(seed),
|
| 26 |
image=control,
|
| 27 |
guess_mode=guess_mode,
|
| 28 |
).images[0]
|
|
@@ -41,7 +40,7 @@ if __name__ == '__main__':
|
|
| 41 |
revision="non-ema",
|
| 42 |
controlnet=controlnet,
|
| 43 |
safety_checker=None).to("cuda")
|
| 44 |
-
|
| 45 |
|
| 46 |
image_types = {'bird', 'human', 'room', 'vermeer'}
|
| 47 |
|
|
@@ -67,9 +66,3 @@ if __name__ == '__main__':
|
|
| 67 |
control=control,
|
| 68 |
guess_mode=True)
|
| 69 |
image.save(f'{output_image_folder}output_{image_type}_{model_name}_{seed}_gm.png')
|
| 70 |
-
# image = generate_image(seed=seed,
|
| 71 |
-
# prompt=test_prompt,
|
| 72 |
-
# negative_prompt=test_negative_prompt,
|
| 73 |
-
# control=control,
|
| 74 |
-
# guess_mode=True)
|
| 75 |
-
# image.save(f'{output_image_folder}output_{image_type}_{model_name}_{seed}_gm_wp.png')
|
|
|
|
| 7 |
import sys
|
| 8 |
import os
|
| 9 |
|
| 10 |
+
from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, DDIMScheduler
|
| 11 |
|
| 12 |
from PIL import Image
|
| 13 |
|
|
|
|
| 19 |
image = pipe(
|
| 20 |
prompt=prompt,
|
| 21 |
negative_prompt=negative_prompt,
|
| 22 |
+
guidance_scale=4.0 if guess_mode else 9.0,
|
| 23 |
+
num_inference_steps=50 if guess_mode else 20,
|
| 24 |
latents=latent,
|
|
|
|
| 25 |
image=control,
|
| 26 |
guess_mode=guess_mode,
|
| 27 |
).images[0]
|
|
|
|
| 40 |
revision="non-ema",
|
| 41 |
controlnet=controlnet,
|
| 42 |
safety_checker=None).to("cuda")
|
| 43 |
+
pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
|
| 44 |
|
| 45 |
image_types = {'bird', 'human', 'room', 'vermeer'}
|
| 46 |
|
|
|
|
| 66 |
control=control,
|
| 67 |
guess_mode=True)
|
| 68 |
image.save(f'{output_image_folder}output_{image_type}_{model_name}_{seed}_gm.png')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
gen_compare_guess_mode/gen_markdown.py
CHANGED
|
@@ -24,6 +24,7 @@ def get_url(path):
|
|
| 24 |
# url = "./output_images/"
|
| 25 |
url = 'https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/'
|
| 26 |
return url + path
|
|
|
|
| 27 |
for cnet in cnet_models:
|
| 28 |
print(f"## ControlNet {cnet}")
|
| 29 |
print("")
|
|
@@ -31,6 +32,9 @@ for cnet in cnet_models:
|
|
| 31 |
|
| 32 |
for image_type in image_types:
|
| 33 |
for seed in range(4):
|
|
|
|
|
|
|
|
|
|
| 34 |
print(f"|{seed}|", end="")
|
| 35 |
for proc in processors:
|
| 36 |
url = get_url(path=f"{proc}/output_{image_type}_{cnet}_{seed:01d}.png")
|
|
|
|
| 24 |
# url = "./output_images/"
|
| 25 |
url = 'https://huggingface.co/takuma104/controlnet_dev/resolve/main/gen_compare_guess_mode/output_images/'
|
| 26 |
return url + path
|
| 27 |
+
|
| 28 |
for cnet in cnet_models:
|
| 29 |
print(f"## ControlNet {cnet}")
|
| 30 |
print("")
|
|
|
|
| 32 |
|
| 33 |
for image_type in image_types:
|
| 34 |
for seed in range(4):
|
| 35 |
+
if cnet=='openpose' and image_type=='human' and seed == 3:
|
| 36 |
+
continue
|
| 37 |
+
|
| 38 |
print(f"|{seed}|", end="")
|
| 39 |
for proc in processors:
|
| 40 |
url = get_url(path=f"{proc}/output_{image_type}_{cnet}_{seed:01d}.png")
|
gen_compare_guess_mode/gen_reference_image.py
CHANGED
|
@@ -63,5 +63,5 @@ if __name__ == '__main__':
|
|
| 63 |
for seed in range(4):
|
| 64 |
image = generate(test_prompt, test_negative_prompt, seed=seed, control=control)
|
| 65 |
image.save(f'{output_image_folder}output_{image_type}_{model_name}_{seed}.png')
|
| 66 |
-
image = generate("", "", seed=seed, control=control, guess_mode=True)
|
| 67 |
image.save(f'{output_image_folder}output_{image_type}_{model_name}_{seed}_gm.png')
|
|
|
|
| 63 |
for seed in range(4):
|
| 64 |
image = generate(test_prompt, test_negative_prompt, seed=seed, control=control)
|
| 65 |
image.save(f'{output_image_folder}output_{image_type}_{model_name}_{seed}.png')
|
| 66 |
+
image = generate("", "", seed=seed, control=control, guess_mode=True, scale=4.0, ddim_steps=50)
|
| 67 |
image.save(f'{output_image_folder}output_{image_type}_{model_name}_{seed}_gm.png')
|
gen_compare_guess_mode/output_images/diffusers/output_bird_canny_0.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_canny_0_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_canny_1.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_canny_1_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_canny_2.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_canny_2_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_canny_3.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_canny_3_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_depth_0.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_depth_0_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_depth_1.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_depth_1_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_depth_2.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_depth_2_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_depth_3.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_depth_3_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_hed_0.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_hed_0_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_hed_1.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_hed_1_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_hed_2.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_hed_2_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_hed_3.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_hed_3_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_mlsd_0.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_mlsd_0_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_mlsd_1.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_mlsd_1_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_mlsd_2.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_mlsd_2_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_mlsd_3.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_mlsd_3_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_normal_0.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_normal_0_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_normal_1.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_normal_1_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_normal_2.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_normal_2_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_normal_3.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_normal_3_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_openpose_0.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_openpose_0_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_openpose_1.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_openpose_1_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_openpose_2.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gen_compare_guess_mode/output_images/diffusers/output_bird_openpose_2_gm.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|