github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-08 20:45:00 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat(nodes): update pidinet node

Browse Source 
Human-readable field names.
This commit is contained in:
psychedelicious
2024-09-11 17:43:55 +10:00
committed by Kent Keirsey
parent a4250e3ff2
commit ee4c0efbf7
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/image_util/pidi/__init__.py
View File

@@ -41,7 +41,7 @@ class PIDINetDetector:
return self
def run(
self, image: Image.Image, safe: bool = False, scribble: bool = False, apply_filter: bool = False
self, image: Image.Image, quantize_edges: bool = False, scribble: bool = False, apply_filter: bool = False
) -> Image.Image:
"""Processes an image and returns the detected edges."""
@@ -62,7 +62,7 @@ class PIDINetDetector:
edge = edge.cpu().numpy()
if apply_filter:
edge = edge > 0.5
if safe:
if quantize_edges:
edge = safe_step(edge)
edge = (edge * 255.0).clip(0, 255).astype(np.uint8)