diff --git a/nodes/image_processing.py b/nodes/image_processing.py index e459c3a..8235913 100644 --- a/nodes/image_processing.py +++ b/nodes/image_processing.py @@ -702,7 +702,6 @@ class MTB_Blur: ) blurred_images.append(blurred) - image_np = np.array(blurred_images) else: for i in range(image.size(0)): blurred = gaussian( @@ -710,8 +709,7 @@ class MTB_Blur: ) blurred_images.append(blurred) - image_np = np.array(blurred_images) - return (np2tensor(image_np).squeeze(0),) + return (np2tensor(blurred_images),) class MTB_Sharpen: diff --git a/nodes/transform.py b/nodes/transform.py index febfe92..381c33d 100644 --- a/nodes/transform.py +++ b/nodes/transform.py @@ -45,6 +45,19 @@ class MTB_TransformImage: ), "constant_color": ("COLOR", {"default": "#000000"}), }, + "optional": { + "filter_type": ( + [ + "nearest", + "box", + "bilinear", + "hamming", + "bicubic", + "lanczos", + ], + {"default": "bilinear"}, + ), + }, } FUNCTION = "transform" @@ -61,7 +74,18 @@ class MTB_TransformImage: shear: float, border_handling="edge", constant_color=None, + filter_type="nearest", ): + filter_map = { + "nearest": Image.NEAREST, + "box": Image.BOX, + "bilinear": Image.BILINEAR, + "hamming": Image.HAMMING, + "bicubic": Image.BICUBIC, + "lanczos": Image.LANCZOS, + } + resampling_filter = filter_map[filter_type] + x = int(x) y = int(y) angle = int(angle) @@ -115,7 +139,12 @@ class MTB_TransformImage: img = cast( Image.Image, TF.affine( - img, angle=angle, scale=zoom, translate=[x, y], shear=shear + img, + angle=angle, + scale=zoom, + translate=[x, y], + shear=shear, + interpolation=resampling_filter, ), ) diff --git a/wiki b/wiki index a402de4..fa7fec2 160000 --- a/wiki +++ b/wiki @@ -1 +1 @@ -Subproject commit a402de4af9cb6d328625496d9931f6b51b9d5225 +Subproject commit fa7fec28a360b43b6c9ae7ee49a29b4c2cc76184