Fix Vignette and Sharpen

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
+3 -3
View File
@@ -31,13 +31,13 @@ class Sharpen:
CATEGORY = "postprocessing/Filters"
def sharpen(self, image: torch.Tensor, blur_radius: int, alpha: float):
if blur_radius == 0:
def sharpen(self, image: torch.Tensor, sharpen_radius: int, alpha: float):
if sharpen_radius == 0:
return (image,)
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
center = kernel_size // 2
kernel[center, center] = kernel_size**2