from PIL import Image
import numpy as np
#为了定义好俯视角el、方位角az与深度,将三个变量定义在前面便于调节
el=np.pi/2.2
az=np.pi/4.
dep=10.
#转化为灰度图
im=Image.open("in.jpg")
im=im.convert('L')
im.save('手绘.jpg')
#取梯度,对图像进行重构
a=np.asarray(im).astype('float')
grd=np.gradient(a)
grd_x,grd_y=grd
grd_x=grd_x*dep/100.
grd_y=grd_y*dep/100.
#制造光源效果
dx=np.cos(el)*np.cos(az)
dy=np.cos(el)*np.sin(az)
dz=np.sin(el)
A = np.sqrt(grd_x**2 + grd_y**2 + 1.)
uni_x=grd_x/A
uni_y=grd_y/A
uni_z=1./A
a2=255*(dx*uni_x+dy*uni_y+dz*uni_z)
a2=a2.clip(0,255)
im0=Image.fromarray(a2.astype('uint8'))
im0.save('手绘.jpg')
其中图片名称需要自定义一下,当然如果和上图一样就无需考虑这么多了。
原图:
运行结果大致如下: