【原】compose组合操作 图片预处理

表白：黑白圣堂血天使，天剑鬼刀阿修罗。 
讲解对象：/compose组合操作 图片预处理
作者：融水公子 rsgz
===
PyTorch技巧  

compose组合操作 图片预处理

from PIL import Image
from torchvision import transforms
from torch.utils.tensorboard import SummaryWriter

pic_one = r"D:\2-code\1-AI_pro\Mangguo_rpo\8-selenium\9-PyTorch\shujuji\hymenoptera_data\train\ants\5650366_e22b7e1065.jpg"
img_pil = Image.open(pic_one)
print(img_pil.size)

# 仅输入一个值就是最小的边和512匹配，可能等比放大也可能缩小
trans_resize = transforms.Resize(512)
trans_totensor = transforms.ToTensor()
# compose 就是把两个函数功能整合在一起 先改变图片大小 然后数据类型转换   等于是多个数据预处理操作联合在了一起
# compose参数 在老版本里面参数顺序必须先resize然后是totensor  新的版本里面没有顺序要求了 新版本resize可以传入tensor型了所以不会报错了
trans_compose = transforms.Compose([trans_resize,trans_totensor])
img_compose = trans_compose(img_pil)

writer = SummaryWriter(r"D:\2-code\1-AI_pro\Mangguo_rpo\8-selenium\9-PyTorch\logs")
writer.add_image("img_compose", img_compose, 1)
writer.close()

显示结果

(500, 375)

tensorboard

=== 

公众号：不浪仙人

谢谢大家的支持！可以点击我的头像，进入我的空间浏览更多文章呢。建议大家360doc[www.360doc.com]注册一个账号登录，里面真的有很多优秀的文章，欢迎大家的到来。
---

';