读取图片并转为numpy数组

from PIL import Image
content_img_path = './style_transfer_data/sun1.jpg'
style_img_path = './style_transfer_data/xingkong.jpeg'

content_val = read_img(content_img_path)
style_val = read_img(style_img_path)


def read_img(img_name):
    img = Image.open(img_name)
    #将读取的图像变为numpy矩阵
    np_img = np.array(img) # (224, 224, 3)
    """
    将3维变为4维矩阵，可以用reshape去做变换，
    因为只有一张图片，可以加一个数组，然后让它封装在一个列表中，
    它就会把列表的这一维也加上去，列表的这一维就是1.
    """
    np_img = np.asarray([np_img], dtype=np.int32) # (1, 224, 224, 3)
    return np_img

注意读取出的图片长宽顺序调换了

PIL读出来的图片size应该是(width,height)，但是转成numpy矩阵后，变成了(height, width, channels)