Commit 3ae61a2b authored by comfyanonymous's avatar comfyanonymous
Browse files

Add a node to invert the colours of images.

Might be useful for controlnets.
parent f04dc2c2
...@@ -727,6 +727,22 @@ class ImageScale: ...@@ -727,6 +727,22 @@ class ImageScale:
s = s.movedim(1,-1) s = s.movedim(1,-1)
return (s,) return (s,)
class ImageInvert:
@classmethod
def INPUT_TYPES(s):
return {"required": { "image": ("IMAGE",)}}
RETURN_TYPES = ("IMAGE",)
FUNCTION = "invert"
CATEGORY = "image"
def invert(self, image):
s = 1.0 - image
return (s,)
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {
"KSampler": KSampler, "KSampler": KSampler,
"CheckpointLoader": CheckpointLoader, "CheckpointLoader": CheckpointLoader,
...@@ -741,6 +757,7 @@ NODE_CLASS_MAPPINGS = { ...@@ -741,6 +757,7 @@ NODE_CLASS_MAPPINGS = {
"LoadImage": LoadImage, "LoadImage": LoadImage,
"LoadImageMask": LoadImageMask, "LoadImageMask": LoadImageMask,
"ImageScale": ImageScale, "ImageScale": ImageScale,
"ImageInvert": ImageInvert,
"ConditioningCombine": ConditioningCombine, "ConditioningCombine": ConditioningCombine,
"ConditioningSetArea": ConditioningSetArea, "ConditioningSetArea": ConditioningSetArea,
"KSamplerAdvanced": KSamplerAdvanced, "KSamplerAdvanced": KSamplerAdvanced,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment