From ee4c0efbf7a1322b1ad7f4f3bc3ab798b86355df Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Wed, 11 Sep 2024 17:43:55 +1000 Subject: [PATCH] feat(nodes): update pidinet node Human-readable field names. --- invokeai/app/invocations/pidi.py | 4 ++-- invokeai/backend/image_util/pidi/__init__.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/invokeai/app/invocations/pidi.py b/invokeai/app/invocations/pidi.py index 1377e7ca66..47b241ee1f 100644 --- a/invokeai/app/invocations/pidi.py +++ b/invokeai/app/invocations/pidi.py @@ -17,7 +17,7 @@ class PiDiNetEdgeDetectionInvocation(BaseInvocation, WithMetadata, WithBoard): """Generates an edge map using PiDiNet.""" image: ImageField = InputField(description="The image to process") - safe: bool = InputField(default=False, description=FieldDescriptions.safe_mode) + quantize_edges: bool = InputField(default=False, description=FieldDescriptions.safe_mode) scribble: bool = InputField(default=False, description=FieldDescriptions.scribble_mode) def invoke(self, context: InvocationContext) -> ImageOutput: @@ -27,7 +27,7 @@ class PiDiNetEdgeDetectionInvocation(BaseInvocation, WithMetadata, WithBoard): with loaded_model as model: assert isinstance(model, PiDiNet) detector = PIDINetDetector(model) - edge_map = detector.run(image=image, safe=self.safe, scribble=self.scribble) + edge_map = detector.run(image=image, quantize_edges=self.quantize_edges, scribble=self.scribble) image_dto = context.images.save(image=edge_map) return ImageOutput.build(image_dto) diff --git a/invokeai/backend/image_util/pidi/__init__.py b/invokeai/backend/image_util/pidi/__init__.py index 99cd1d5e6f..8673b21914 100644 --- a/invokeai/backend/image_util/pidi/__init__.py +++ b/invokeai/backend/image_util/pidi/__init__.py @@ -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)