python手绘

 1 from PIL import Image
 2 import numpy as np
 3      
 4 a = np.asarray(Image.open(r‘C:\Users\ldy\Desktop\214.jpg‘).convert(‘L‘)).astype(‘float‘)
 5      
 6 depth = 10.                      # (0-100)
 7 grad = np.gradient(a)             #取图像灰度的梯度值，表示图像灰度的变化率
 8 grad_x, grad_y = grad               #分别取横纵图像梯度值
 9 grad_x = grad_x*depth/100.
10 grad_y = grad_y*depth/100.          #根据深度来调整像素的梯度值，从而调整明暗程度
11 A = np.sqrt(grad_x**2 + grad_y**2 + 1.)#构造x轴和y轴梯度的三维归一化单位坐标系
12 uni_x=grad_x/A
13 uni_y = grad_y/A
14 uni_z = 1./A
15      
16 vec_el = np.pi/2.2                   # 光源的俯视角度，弧度值
17 vec_az = np.pi/4.                    # 光源的方位角度，弧度值
18 dx = np.cos(vec_el)*np.cos(vec_az)   #光源对x 轴的影响
19 dy = np.cos(vec_el)*np.sin(vec_az)   #光源对y 轴的影响
20 dz = np.sin(vec_el)              #光源对z 轴的影响
21      
22 b = 255*(dx*uni_x + dy*uni_y + dz*uni_z)     #梯度和光源相互作用，将梯度转化为灰度，光源归一化
23 b = b.clip(0,255)#避免数据越界
24      
25 im = Image.fromarray(b.astype(‘uint8‘))  #重构图像
26 im.save(r‘C:\Users\ldy\Desktop\50.jpg‘)