From 2451f43fa7322b2d82cf61e74c51bb202acaa85c Mon Sep 17 00:00:00 2001
From: AngelBottomless <aria1th@naver.com>
Date: Fri, 7 Feb 2025 15:57:06 +0900
Subject: [PATCH] Use pixelate with max tiles

---
 auxilary.py | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/auxilary.py b/auxilary.py
index 5dab34f..660eaf5 100644
--- a/auxilary.py
+++ b/auxilary.py
@@ -59,6 +59,22 @@ def pixelate(image, pixelation_factor=0.1):
     # Upscale back to original size
     return small.resize(image.size, Image.NEAREST)
 
+
+def pixelate_target_tiles(image, max_tiles=100):
+    aspect_ratio = image.width / image.height
+    # solve h * a * h = max_tiles
+    h = int((max_tiles / aspect_ratio) ** 0.5)
+    w = int(aspect_ratio * h)
+    h, w = max(4, h), max(4, w)
+    # Downscale the image
+    small = image.resize(
+        (w, h),
+        resample=Image.NEAREST,
+    )
+    # Upscale back to original size
+    return small.resize(image.size, Image.NEAREST)
+
+
 @auxilary_node
 class CensorImageByRating:
     FUNCTION = "censor_image"
@@ -104,7 +120,7 @@ class CensorImageByRating:
             elif censor_method.lower() == "pixelate":
                 # first, blur
                 censored_image = image.filter(ImageFilter.GaussianBlur(radius=20))
-                censored_image = pixelate(censored_image, 0.05)
+                censored_image = pixelate_target_tiles(censored_image, max_tiles=100)
                 return (censored_image,)
 
         # If unknown method is provided, just return the original image