Python Pillow 图像处理与格式转换详解

5.2.2 图像模式与转换

理解图像模式至关重要，不同的模式对应不同的颜色空间和像素表示。Pillow 支持多种模式，从简单的二值图到复杂的 LAB 空间。

下面的代码展示了如何进行模式间的灵活转换，特别是处理透明度和灰度时的注意事项：

from PIL import Image
import numpy as np

class ImageModeConverter:
    def __init__(self, image: Image.Image):
        self.image = image
        self.original_mode = image.mode

    def to_grayscale(self, method='standard') -> Image.Image:
        if method == 'standard':
            return self.image.convert('L')
        rgb = self.image.convert('RGB')
        arr = np.array(rgb, dtype=np.float64)
        # 使用亮度公式计算灰度
        gray = 0.299 * arr[:,:,0] + 0.587 * arr[:,:,1] + 0.114 * arr[:,:,2]
        return Image.fromarray(gray.astype(np.uint8), mode='L')

    def to_rgba(self, background_color=(255, 255, 255)) -> Image.Image:
        if self.image.mode == 'RGBA':
            return self.image.copy()
        # 注意：非 RGBA 转 RGBA 需要指定背景色填充透明通道
        return self.image.convert('RGBA')

    def to_binary(self, threshold=128) -> Image.Image:
        gray = self.image.convert('L')
        return gray.point(lambda x: 255 if x > threshold else 0, mode='1')

5.3 图像基本操作

5.3.1 几何变换

Pillow 提供了丰富的几何变换功能，包括裁剪、缩放、旋转、翻转等。这些操作默认返回新的 Image 对象，不会修改原始数据。在实际应用中，我们常需要根据目标尺寸自适应调整图片。

from PIL import Image, ImageOps

class PillowGeometry:
    def __init__(self, image: Image.Image):
        self.image = image
        self.width, self.height = image.size

    def resize_contain(self, size: tuple, background_color=(255, 255, 255)) -> Image.Image:
        """缩放并包含在指定尺寸内（保持宽高比，不足部分填充）"""
        return ImageOps.fit(self.image, size, method=Image.Resampling.LANCZOS, bleed=0.0, centering=(0.5, 0.5))

    def resize_cover(self, size: tuple) -> Image.Image:
        """缩放并覆盖指定尺寸（保持宽高比，可能裁剪）"""
        ratio = max(size[0] / self.width, size[1] / self.height)
        new_size = (int(self.width * ratio), int(self.height * ratio))
        resized = self.image.resize(new_size, resample=Image.Resampling.LANCZOS)
        left = (resized.width - size[0]) // 2
        upper = (resized.height - size[1]) // 2
        return resized.crop((left, upper, left + size[0], upper + size[1]))

    def create_thumbnail(self, size: tuple) -> Image.Image:
        img_copy = self.image.copy()
        img_copy.thumbnail(size, resample=Image.Resampling.LANCZOS)
        return img_copy

5.3.2 图像增强

通过 ImageEnhance 和 ImageFilter 模块，我们可以轻松调整图像的视觉效果。无论是提升清晰度还是添加艺术滤镜，Pillow 都能胜任。

from PIL import ImageEnhance, ImageFilter

class PillowEnhance:
    def __init__(self, image: Image.Image):
        self.image = image.convert('RGB')

    def adjust_brightness(self, factor: float) -> Image.Image:
        enhancer = ImageEnhance.Brightness(self.image)
        return enhancer.enhance(factor)

    def apply_filter(self, filter_type: str, **kwargs) -> Image.Image:
        filters = {
            'blur': ImageFilter.BLUR,
            'sharpen': ImageFilter.SHARPEN,
            'emboss': ImageFilter.EMBOSS,
            'gaussian': ImageFilter.GaussianBlur(radius=kwargs.get('radius', 2))
        }
        if filter_type in filters:
            return self.image.filter(filters[filter_type])
        return self.image.copy()

5.4 图像绘制与文字处理

5.4.1 图像绘制

ImageDraw 模块允许我们在图像上绘制直线、矩形、圆形、多边形以及文字。这对于生成标注图、水印或简单图表非常有用。

from PIL import ImageDraw, ImageFont

class PillowDrawing:
    def __init__(self, image=None, size=(640, 480), background='white'):
        if image is not None:
            self.image = image.copy()
        else:
            self.image = Image.new('RGB', size, background)
        self.draw = ImageDraw.Draw(self.image)
        self.width, self.height = self.image.size

    def draw_text_centered(self, text: str, color='black', font=None):
        bbox = self.draw.textbbox((0, 0), text, font=font)
        text_width = bbox[2] - bbox[0]
        text_height = bbox[3] - bbox[1]
        x = (self.width - text_width) // 2
        y = (self.height - text_height) // 2
        self.draw.text((x, y), text, fill=color, font=font)

    def draw_arrow(self, start: tuple, end: tuple, color='black', width=2):
        self.draw.line([start, end], fill=color, width=width)
        # 这里可以进一步计算箭头头部顶点绘制多边形

5.5 本章小结

本章详细介绍了 Pillow 库的核心用法，涵盖了从基础的文件读写、格式转换，到进阶的几何变换、色彩增强及绘图功能。Pillow 凭借其简洁的 API 和广泛的格式支持，成为 Python 图像处理的首选工具之一。

虽然 OpenCV 在计算机视觉领域更为强大，但 Pillow 在处理静态图像、UI 素材生成及格式转换方面具有不可替代的优势。两者结合使用往往能发挥最大效能。下一章我们将深入探讨色彩空间转换与通道操作的底层原理。