使用Python实现图像的手绘风格效果

一、引言

在数字图像处理领域，模拟手绘风格是一项有趣且具有挑战性的任务。手绘风格图像通常具有独特的纹理和深浅变化，给人一种亲切和艺术感。本文旨在通过计算图像的梯度并模拟光照效果，实现一种简单而有效的手绘风格图像生成方法。

二、代码详细解释与示例

1. 导入必要的库：

   from PIL import Image
   import numpy as np
   import os
   

   • PIL（Pillow）：用于图像的打开、转换和保存。
   • NumPy：用于数组和矩阵操作，以及数学计算。
   • os：用于文件和目录操作。

2. 计算图像的阴影效果：

    def calculate_shading(img, depth, vec_el, vec_az):
        # 计算图像在x和y方向上的梯度
        grad = np.gradient(img)
        grad_x, grad_y = grad  # grad_x是x方向的梯度，grad_y是y方向的梯度
    
        # 根据深度因子调整梯度大小
        grad_x *= depth / 100.
        grad_y *= depth / 100.
    
        # 计算梯度的模（大小），并添加一个小常数以避免除以零
        A = np.sqrt(grad_x ** 2 + grad_y ** 2 + 1e-7)  # 1e-7是为了数值稳定性
    
        # 计算单位梯度向量（在x, y, z方向上的分量）
        # 这里我们假设z方向上的分量为1/A，因为它代表图像表面的“法线”方向
        uni_x = grad_x / A
        uni_y = grad_y / A
        uni_z = 1. / A
    
        # 计算光源方向向量（在x, y, z方向上的分量）
        dx = np.cos(vec_el) * np.cos(vec_az)  # 光源在x方向的影响
        dy = np.cos(vec_el) * np.sin(vec_az)  # 光源在y方向的影响
        dz = np.sin(vec_el)                   # 光源在z方向（垂直于图像平面）的影响
    
        # 计算光源方向与单位梯度向量的点积，模拟光照效果
        # 点积越大，表示该点越亮；点积越小（越接近-1），表示该点越暗（但由于我们取绝对值并映射到0-255，所以实际上表现为更深的阴影）
        # 但由于我们是模拟手绘风格，所以直接乘以255并裁剪到0-255范围内
        b = 255 * (dx * uni_x + dy * uni_y + dz * uni_z)