本文纯属自己实验记录,欢迎大家共同交流学习,顺利毕业!
对RGB图片进行滤波,因为RGB是三维数组,类型为HWC,所以本文仅采用numpy、PIL库对图片进行中值滤波。
首先,RGB图片转换成np.array。
#导入相应的库
from PIL import Image
import numpy as np
image = Image.open(in_file)#打开rgb图片
img = np.array(image)#转换成数组
其次,将三维数组(三个通道)分别提取出来,作为3个2维数组。
img_red = img[:, :, 0]
img_green = img[:, :, 1]
img_blue = img[:, :, 2]
接着,获取图片的宽度与高度,在滤波循环中使用。由于对图片进行中值滤波之后,图片的像素值发生改变,我们需要创建一个新的空数组存放我们处理过的图片。
height, width = img_red.shape
new_array_red = np.zeros((height, width), dtype=int)
new_array_green = np.zeros((height, width), dtype=int)
new_array_blue = np.zeros((height, width), dtype=int)
接着,对图片进行中值滤波。pad为滤波器的半径(不知道理解对不对),i与j的范围需要理解,从哪里开始,我的i与j是从图片的左上角开始,也就是滤波器正好左上角与图片左上角重合。如果滤波器左上角范围在图片外面,range范围需要修