（numpy+image）获取手绘效果的图片

原理就不说了，自己也没怎么弄明白，直接上代码，有兴趣的话直接看看视频吧。

import numpy as np
from PIL import Image

a = np.asarray(Image.open(r'C:\Users\pc\Pictures\Saved Pictures\桌面.jpg').convert(('L'))).astype('float')

depth = 10
grad = np.gradient(a)
grad_x, grad_y = grad
grad_x = grad_x*depth/100
grad_y = grad_y*depth/100

vec_el = np.pi/2.2    # 光源的俯视角度，弧度值
vec_az = np.pi/4.    # 光源的方位角度，弧度值
dx = np.cos(vec_el) * np.cos(vec_az)
dy = np.cos(vec_el) * np.sin(vec_az)
dz = np.sin(vec_el)

A = np.sqrt(grad_x**2+grad_y**2+1.)
# 光源对x，y，z轴的影响
uni_x = grad_x/A
uni_y = grad_y/A
uni_z = 1./A

# 光源归一化
b = 255*(dx*uni_x + dy*uni_y + dz*uni_z)
b = b.clip(0,255)

# 重构图像并保存
im = Image.fromarray(b.astype('uint8'))
im.save(r'C:\Users\pc\Desktop\03.jpg')

附上视频的地址