nodes/EllangoK-ComfyUI-post-processing-nodes
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix Vignette and Sharpen

Browse Source
This commit is contained in:
EllangoK
2024-08-09 12:50:47 -04:00
parent d5bacf7925
commit 522817d3d3
3 changed files with 1474 additions and 1474 deletions
Download Patch File Download Diff File Expand all files Collapse all files
post_processing/sharpen.py
+3 -3
View File
@@ -31,13 +31,13 @@ class Sharpen:
CATEGORY = "postprocessing/Filters" CATEGORY = "postprocessing/Filters"
def sharpen(self, image: torch.Tensor, blur_radius: int, alpha: float): def sharpen(self, image: torch.Tensor, sharpen_radius: int, alpha: float):
if blur_radius == 0: if sharpen_radius == 0:
return (image,) return (image,)
batch_size, height, width, channels = image.shape batch_size, height, width, channels = image.shape
kernel_size = blur_radius * 2 + 1 kernel_size = sharpen_radius * 2 + 1
kernel = torch.ones((kernel_size, kernel_size), dtype=torch.float32) * -1 kernel = torch.ones((kernel_size, kernel_size), dtype=torch.float32) * -1
center = kernel_size // 2 center = kernel_size // 2
kernel[center, center] = kernel_size**2 kernel[center, center] = kernel_size**2
post_processing/vignette.py
+6 -6
View File
@@ -10,11 +10,11 @@ class Vignette:
return { return {
"required": { "required": {
"image": ("IMAGE",), "image": ("IMAGE",),
"a": ("FLOAT", { "vignette": ("FLOAT", {
"default": 0.0, "default": 0.0,
"min": 0.0, "min": 0.0,
"max": 10.0, "max": 10.0,
"step": 1.0 "step": 0.01
}), }),
}, },
} }
@@ -33,10 +33,10 @@ class Vignette:
X, Y = torch.meshgrid(x, y, indexing="ij") X, Y = torch.meshgrid(x, y, indexing="ij")
radius = torch.sqrt(X ** 2 + Y ** 2) radius = torch.sqrt(X ** 2 + Y ** 2)
# Map vignette strength from 0-10 to 1.800-0.800 radius = radius / torch.amax(radius, dim=(0, 1), keepdim=True)
mapped_vignette_strength = 1.8 - (vignette - 1) * 0.1 opacity = torch.tensor(vignette, device=image.device)
vignette = 1 - torch.clamp(radius / mapped_vignette_strength, 0, 1) opacity = torch.clamp(opacity, 0.0, 1.0)
vignette = vignette[..., None] vignette = 1 - radius.unsqueeze(0).unsqueeze(-1) * opacity
vignette_image = torch.clamp(image * vignette, 0, 1) vignette_image = torch.clamp(image * vignette, 0, 1)
post_processing_nodes.py
+9 -9
View File
@@ -1195,13 +1195,13 @@ class Sharpen:
CATEGORY = "postprocessing/Filters" CATEGORY = "postprocessing/Filters"
def sharpen(self, image: torch.Tensor, blur_radius: int, alpha: float): def sharpen(self, image: torch.Tensor, sharpen_radius: int, alpha: float):
if blur_radius == 0: if sharpen_radius == 0:
return (image,) return (image,)
batch_size, height, width, channels = image.shape batch_size, height, width, channels = image.shape
kernel_size = blur_radius * 2 + 1 kernel_size = sharpen_radius * 2 + 1
kernel = torch.ones((kernel_size, kernel_size), dtype=torch.float32) * -1 kernel = torch.ones((kernel_size, kernel_size), dtype=torch.float32) * -1
center = kernel_size // 2 center = kernel_size // 2
kernel[center, center] = kernel_size**2 kernel[center, center] = kernel_size**2
@@ -1309,11 +1309,11 @@ class Vignette:
return { return {
"required": { "required": {
"image": ("IMAGE",), "image": ("IMAGE",),
"a": ("FLOAT", { "vignette": ("FLOAT", {
"default": 0.0, "default": 0.0,
"min": 0.0, "min": 0.0,
"max": 10.0, "max": 10.0,
"step": 1.0 "step": 0.01
}), }),
}, },
} }
@@ -1332,10 +1332,10 @@ class Vignette:
X, Y = torch.meshgrid(x, y, indexing="ij") X, Y = torch.meshgrid(x, y, indexing="ij")
radius = torch.sqrt(X ** 2 + Y ** 2) radius = torch.sqrt(X ** 2 + Y ** 2)
# Map vignette strength from 0-10 to 1.800-0.800 radius = radius / torch.amax(radius, dim=(0, 1), keepdim=True)
mapped_vignette_strength = 1.8 - (vignette - 1) * 0.1 opacity = torch.tensor(vignette, device=image.device)
vignette = 1 - torch.clamp(radius / mapped_vignette_strength, 0, 1) opacity = torch.clamp(opacity, 0.0, 1.0)
vignette = vignette[..., None] vignette = 1 - radius.unsqueeze(0).unsqueeze(-1) * opacity
vignette_image = torch.clamp(image * vignette, 0, 1) vignette_image = torch.clamp(image * vignette, 0, 1)