Update pixelation.py

pixelation.py

@@ -14,14 +14,10 @@ def pixelate_image(image, pixel_size, interpolation="Nearest"):
     返回:
     - 像素化后的 PIL 图像对象
     """
-    # 将输入图像转为 RGB 模式
     img = image.convert("RGB")
-
-    # 获取原图像的尺寸
     width, height = img.size
-    # pixel_size = max(1, round(min(width, height) / 1024) * pixel_size)
+    pixel_size = max(1, round(min(width, height) / 1024) * pixel_size)
 
-    # 选择插值方式
     if interpolation == "Nearest":
         resample_method = Image.NEAREST
     elif interpolation == "Bilinear":
@@ -33,13 +29,11 @@ def pixelate_image(image, pixel_size, interpolation="Nearest"):
     else:
         raise ValueError(f"未知的插值方法: {interpolation}")
 
-    # 第一步：缩小图像，使用邻近插值保持像素块的正方形效果
     small_img = img.resize(
         (width // pixel_size, height // pixel_size),
         resample=resample_method
     )
 
-    # 第二步：放大图像，使用用户选择的插值方法
     pixelated_img = small_img.resize(
         (width, height),
         resample=resample_method
@@ -47,7 +41,6 @@ def pixelate_image(image, pixel_size, interpolation="Nearest"):
 
     return pixelated_img
 
-
 def mosaic_pixelation(image, pixel_size):
     """
     使用马赛克方法对图像进行像素化。
@@ -62,7 +55,7 @@ def mosaic_pixelation(image, pixel_size):
     img = image.convert("RGB")
     img_np = np.array(img)
     h, w, _ = img_np.shape
-    pixel_size = round(min(w, h) / 1024) * pixel_size
+    pixel_size = max(1, round(min(w, h) / 1024) * pixel_size)
 
     for y in range(0, h, pixel_size):
         for x in range(0, w, pixel_size):
@@ -72,7 +65,6 @@ def mosaic_pixelation(image, pixel_size):
 
     return Image.fromarray(img_np)
 
-
 def oil_paint_pixelation(image, pixel_size):
     """
     使用油画滤镜方法对图像进行像素化。
@@ -87,7 +79,8 @@ def oil_paint_pixelation(image, pixel_size):
     img = image.convert("RGB")
     img_np = np.array(img)
     h, w, _ = img_np.shape
-    pixel_size = round(min(w, h) / 1024) * pixel_size
+    pixel_size = max(1, round(min(w, h) / 1024) * pixel_size)
+
     for y in range(0, h, pixel_size):
         for x in range(0, w, pixel_size):
             block = img_np[y:y + pixel_size, x:x + pixel_size]
@@ -97,7 +90,6 @@ def oil_paint_pixelation(image, pixel_size):
 
     return Image.fromarray(img_np)
 
-
 def hierarchical_pixelation(image, min_pixel_size, max_pixel_size):
     """
     使用层次像素化方法对图像进行像素化。
@@ -113,8 +105,8 @@ def hierarchical_pixelation(image, min_pixel_size, max_pixel_size):
     img = image.convert("RGB")
     img_np = np.array(img)
     h, w, _ = img_np.shape
-    min_pixel_size = round(min(w, h) / 1024) * min_pixel_size
-    max_pixel_size = round(min(w, h) / 1024) * max_pixel_size
+    min_pixel_size = max(1, round(min(w, h) / 1024) * min_pixel_size)
+    max_pixel_size = max(1, round(min(w, h) / 1024) * max_pixel_size)
 
     step = max((max_pixel_size - min_pixel_size) // (w // min_pixel_size), 1)