diff --git a/.gitignore b/.gitignore index a065928..de2b903 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,2 @@ __pycache__ -images/ +images/.ref/ diff --git a/README.md b/README.md index e69de29..546d406 100644 --- a/README.md +++ b/README.md @@ -0,0 +1,60 @@ +An extension for [stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui) that adds a custom script which let you to observe U-net feature maps. + +# Example + +Model Output Image: + +![Model Output Image](images/00.png) + +``` +Model: waifu-diffusion-v1-3-float16 (84692140) +Prompt: a cute girl, pink hair +Sampling Method: DPM++ 2M Karras +Size: 512x512 +CFG Scale: 7 +Seed: 1719471015 +``` + +U-net features: + +- IN00 (64x64, 320ch) + +Let the feature value is `v`, larger `|v|` is white, and zero is black. + +step 1 + +![IN00 step1](images/IN00-step01.png) + +step 10 + +![IN00 step10](images/IN00-step10.png) + +step 20 + +![IN00 step20](images/IN00-step20.png) + +- OUT02 (16x16, 1280ch) + +step 20 + +![OUT02 step20](images/OUT02-step20.png) + +- OUT11 (64x64, 320ch) + +step 1 + +![OUT11 step1](images/OUT11-step01.png) + +step 10 + +![OUT11 step10](images/OUT11-step10.png) + +step 20 + +![OUT11 step20](images/OUT11-step20.png) + +Color map mode: + +Red means the value is positive, and blue means the value is negative. + +![OUT11 step20 cm](images/OUT11-step20-cm.png) diff --git a/images/00.png b/images/00.png new file mode 100644 index 0000000..69a3e34 Binary files /dev/null and b/images/00.png differ diff --git a/images/IN00-step01.png b/images/IN00-step01.png new file mode 100644 index 0000000..38a4115 Binary files /dev/null and b/images/IN00-step01.png differ diff --git a/images/IN00-step10.png b/images/IN00-step10.png new file mode 100644 index 0000000..3020025 Binary files /dev/null and b/images/IN00-step10.png differ diff --git a/images/IN00-step20.png b/images/IN00-step20.png new file mode 100644 index 0000000..73407bf Binary files /dev/null and b/images/IN00-step20.png differ diff --git a/images/OUT02-step20.png b/images/OUT02-step20.png new file mode 100644 index 0000000..7ddbc2c Binary files /dev/null and b/images/OUT02-step20.png differ diff --git a/images/OUT11-step01-cm.png b/images/OUT11-step01-cm.png new file mode 100644 index 0000000..787a57b Binary files /dev/null and b/images/OUT11-step01-cm.png differ diff --git a/images/OUT11-step01.png b/images/OUT11-step01.png new file mode 100644 index 0000000..7c93bf4 Binary files /dev/null and b/images/OUT11-step01.png differ diff --git a/images/OUT11-step10-cm.png b/images/OUT11-step10-cm.png new file mode 100644 index 0000000..922247f Binary files /dev/null and b/images/OUT11-step10-cm.png differ diff --git a/images/OUT11-step10.png b/images/OUT11-step10.png new file mode 100644 index 0000000..b95017b Binary files /dev/null and b/images/OUT11-step10.png differ diff --git a/images/OUT11-step20-cm.png b/images/OUT11-step20-cm.png new file mode 100644 index 0000000..d84813f Binary files /dev/null and b/images/OUT11-step20-cm.png differ diff --git a/images/OUT11-step20.png b/images/OUT11-step20.png new file mode 100644 index 0000000..7ab3b6b Binary files /dev/null and b/images/OUT11-step20.png differ