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)