python将照片转换成手绘图片

最近在b站学习数据分析时，看到了可以用numpy将照片转换成手绘效果。自己整上代码学习试了半天，调整了参数并看了效果。代码如下：

# -*- coding: utf-8 -*-
"""
Spyder Editor

This is a temporary script file.
"""

from PIL import Image
import numpy as np

a = np.array(Image.open(D:/Download/20200317194151.jpg).convert(L)).astype(float)

print(a)
depth = 10.
grad = np.gradient(a)  #取图像梯度值
grad_x,grad_y = grad  #分别取横纵梯度值
grad_x = grad_x*depth/30.#根据想的得到的效果自己调节
grad_y = grad_y*depth/30.#根据想的得到的效果自己调节
A = np.sqrt(grad_x**2 + grad_y **2 +1.)
uni_x = grad_x/A
uni_y = grad_y/A
uni_z = 1./A

vec_el = np.pi/2                     #光源俯视角度
vec_az = np.pi/4.                    #光源的方位角度
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轴的影响

b= 255*(dx*uni_x + dy*uni_y + dz*uni_z)  #光源归一化
b=b.clip(0,255)
im = Image.fromarray(b.astype(uint8))#将照片重新定义
im.save(D:/Download/20200317.jpg)#保存

转换前的图片是： 转换后的照片样子效果如下： 之后再学习一下将照片像素调高的操作方法