from PIL import Image import numpy as np #封装一个图像处理类 class TestNumpy(object): def photo2paint(self,img_url): # 读取图片 my_photo = np.asarray(Image.open(img_url).convert("L")).astype("float") # print(my_photo.shape) #设置灰度阈值(范围0~100) depth = 40 # 将灰度系数映射到numpy grad = np.gradient(my_photo) #获取坐标 grad_x,grad_y = grad #分别处理 grad_x = grad_x * depth/100 grad_y = grad_y * depth/100 #设置阿尔法值 设置坐标范围 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.2 vec_az = np.pi/4 #设置x轴和y轴 dx = np.cos(vec_el)* np.cos(vec_az) dy = np.cos(vec_el)* np.sin(vec_az) #设置z轴 dz = np.sin(vec_el) #设置byte值 b = 255*(dx*uni_x+dy*uni_y+dz*uni_z) #光源归一化 b = b.clip(0,255) #声明图像类图像 im = Image.fromarray(b.astype("uint8")) #保存图像 im.save("./test_new.jpg") if __name__ == "__main__": testnumpy = TestNumpy() testnumpy.photo2paint('./test_numpy.jpg')
