Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: extract attribute canny_high_threshold

Browse Source
This commit is contained in:
Manuel Schmid
2024-01-22 19:08:54 +01:00
parent ec486443ea
commit 2d8ca41ce5
4 changed files with 16 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/async_worker.py
+2 -1
View File
@@ -155,6 +155,7 @@ def worker():
debugging_cn_preprocessor = args.pop()
skipping_cn_preprocessor = args.pop()
canny_low_threshold = args.pop()
canny_high_threshold = args.pop()
cn_tasks = {x: [] for x in flags.ip_list}
for _ in range(4):
@@ -647,7 +648,7 @@ def worker():
cn_img = resize_image(HWC3(cn_img), width=width, height=height)
if not skipping_cn_preprocessor:
cn_img = preprocessors.canny_pyramid(cn_img, canny_low_threshold)
cn_img = preprocessors.canny_pyramid(cn_img, canny_low_threshold, canny_high_threshold)
cn_img = HWC3(cn_img)
task[0] = core.numpy_to_pytorch(cn_img)