Skip to content

GitLab

"...kernels/git@developer.sourcefind.cn:OpenDAS/lmdeploy.git" did not exist on "e432dbb0e56caaf319b9c9d7b79eb8106852dc91"
Commit 871a76b7 authored by comfyanonymous's avatar comfyanonymous
Browse files

Rename and reorganize post processing nodes.

parent 10ad4c1d
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 8 deletions
comfy_extras/nodes_post_processing.py
View file @ 871a76b7
...@@ -29,7 +29,7 @@ class Blend: ...@@ -29,7 +29,7 @@ class Blend:
RETURN_TYPES = ("IMAGE",) RETURN_TYPES = ("IMAGE",)
FUNCTION = "blend_images" FUNCTION = "blend_images"
CATEGORY = "postprocessing" CATEGORY = "image/postprocessing"
def blend_images(self, image1: torch.Tensor, image2: torch.Tensor, blend_factor: float, blend_mode: str): def blend_images(self, image1: torch.Tensor, image2: torch.Tensor, blend_factor: float, blend_mode: str):
if image1.shape != image2.shape: if image1.shape != image2.shape:
...@@ -86,7 +86,7 @@ class Blur: ...@@ -86,7 +86,7 @@ class Blur:
RETURN_TYPES = ("IMAGE",) RETURN_TYPES = ("IMAGE",)
FUNCTION = "blur" FUNCTION = "blur"
CATEGORY = "postprocessing" CATEGORY = "image/postprocessing"
def gaussian_kernel(self, kernel_size: int, sigma: float): def gaussian_kernel(self, kernel_size: int, sigma: float):
x, y = torch.meshgrid(torch.linspace(-1, 1, kernel_size), torch.linspace(-1, 1, kernel_size), indexing="ij") x, y = torch.meshgrid(torch.linspace(-1, 1, kernel_size), torch.linspace(-1, 1, kernel_size), indexing="ij")
...@@ -131,7 +131,7 @@ class Quantize: ...@@ -131,7 +131,7 @@ class Quantize:
RETURN_TYPES = ("IMAGE",) RETURN_TYPES = ("IMAGE",)
FUNCTION = "quantize" FUNCTION = "quantize"
CATEGORY = "postprocessing" CATEGORY = "image/postprocessing"
def quantize(self, image: torch.Tensor, colors: int = 256, dither: str = "FLOYDSTEINBERG"): def quantize(self, image: torch.Tensor, colors: int = 256, dither: str = "FLOYDSTEINBERG"):
batch_size, height, width, _ = image.shape batch_size, height, width, _ = image.shape
...@@ -179,7 +179,7 @@ class Sharpen: ...@@ -179,7 +179,7 @@ class Sharpen:
RETURN_TYPES = ("IMAGE",) RETURN_TYPES = ("IMAGE",)
FUNCTION = "sharpen" FUNCTION = "sharpen"
CATEGORY = "postprocessing" CATEGORY = "image/postprocessing"
def sharpen(self, image: torch.Tensor, sharpen_radius: int, alpha: float): def sharpen(self, image: torch.Tensor, sharpen_radius: int, alpha: float):
if sharpen_radius == 0: if sharpen_radius == 0:
...@@ -203,8 +203,8 @@ class Sharpen: ...@@ -203,8 +203,8 @@ class Sharpen:
return (result,) return (result,)
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {
"Blend": Blend, "ImageBlend": Blend,
"Blur": Blur, "ImageBlur": Blur,
"Quantize": Quantize, "ImageQuantize": Quantize,
"Sharpen": Sharpen, "ImageSharpen": Sharpen,
} }
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