PIL给图片加掩膜

Image图片

Image图片是PIL库的图片

from PIL import Image
img = Image.open(path) #打开图片
img = Image.resize((target_H, target_W)) #resize操作
img.save(path) #保存图片

Image图像转换成numpy array数据类型

img = Image.open(path) #打开图片
array = np.array(img)

给图片加掩膜

掩膜是二维的array数组，有膜的像素值为0，无膜的像素值为1
将掩膜与原图相乘即可为图像加掩膜

img = Image.open(path) #打开图片
mask = sio.loadmet(path) #打开掩膜
array = np.array(img)
# array维度 [W, H, C] -> [C, W, H]
array = np.transpose(array, [2, 0, 1])
array = array * mask #点乘
# array维度 [C, W, H] -> [W, H, C] 
array = np.transpose(array, [1, 2, 0])

numpy array转换成Image图像

img = Image.fromarray(array, mode='RGB') #打开图片

Image图像显示

img.show()