在前文 Python 绘图生成 Label Mask 的同时, 需要从文件目录读取文件并再一次压缩或扩展至指定大小。以此达到绘制 Label Mask 的目的。
依赖
import numpy as np
import matplotlib
import PIL
from matplotlib import pyplot as plt
代码
以下代码展示读取 file_name
灰阶图片文件(或彩色)并转换到 $256 \times 256$ 大小的 numpy 数组格式中。
image = np.sum(matplotlib.image.imread(fileName),2)
image = 1-(image-1)/3
image = PIL.Image.fromarray(image*255).convert('L')
image = image.resize((256, 256))
image = np.asarray(image)/255