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
|