lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-05 13:09:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update sd_forge_controlnet_example.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-27 16:48:04 -08:00
parent 7e0c454dca
commit 5d3071331a
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
extensions-builtin/sd_forge_controlnet_example/scripts/sd_forge_controlnet_example.py
View File

@@ -52,11 +52,15 @@ class ControlNetExampleForge(scripts.Script):
print('The model [control_v11p_sd15_canny_fp16.safetensors] download finished.')
controlnet = load_controlnet(controlnet_canny_path)
print('Controlnet loaded.')
input_image = cv2.resize(input_image, (p.height, p.width))
canny_image = cv2.Canny(input_image, 100, 200)
# Display preprocessor result. Called every batch. Cache in your own way.
p.extra_result_images.append(input_image)
p.extra_result_images.append(canny_image)
print('Preprocessor Canny finished.')
return