常见的Transforms（二）（Resize）

Resize

代码

# resize
print(img.size)
trans_resize = transforms.Resize((128, 128))
# 经过大小变换后的img还是PIL数据类型的，还需要将它转换位TOTENSOR类型的
img_resize = trans_resize(img)
img_resize = trans_totensor(img_resize)
print(img_resize)

writer.add_image("Resize", img_resize, 0)

• 将图片的大小进行转换
• 再次转换位totensor进行输出

结果

图片大小发生改变

使用compose-》resize

compose用法

第一个的输出是第二个输入

代码

# compose - resize -2
trans_resize_2 = transforms.Resize(128)
# 将工具箱中的工具拿出来
trans_compose = transforms.Compose([trans_resize_2, trans_totensor])
img_resize_2 = trans_compose(img)
writer.add_image("Resize", img_resize_2, 1)

结果

出现step0和1

注意一点！！

两个参数

• 后面参数的需要的输入数据应该是前面参数的输出结果