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
注意一点!!
两个参数
后面参数的需要的输入
数据应该是前面参数的输出结果