表白:黑白圣堂血天使,天剑鬼刀阿修罗。 讲解对象:/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)
# 裁剪的参数 需要注意 如果 比原图还大就报错 128通常不会报错 因为已经很小了
trans_randcrop = transforms.RandomCrop(128)
trans_totensor = transforms.ToTensor()
# compose 就是把两个函数功能整合在一起 先改变图片大小 然后数据类型转换 等于是多个数据预处理操作联合在了一起
# compose参数 在老版本里面参数顺序必须先randcrop然后是totensor 新的版本里面没有顺序要求了 新版本可以传入tensor型了所以不会报错了
trans_compose = transforms.Compose([trans_randcrop,trans_totensor])
img_compose = trans_compose(img_pil)
writer = SummaryWriter(r"D:\2-code\1-AI_pro\Mangguo_rpo\8-selenium\9-PyTorch\logs")
for i in range(10):
writer.add_image("img_compose", img_compose, i)
writer.close() tensorboard结果
=== 公众号:不浪仙人 谢谢大家的支持!可以点击我的头像,进入我的空间浏览更多文章呢。建议大家360doc[www.360doc.com]注册一个账号登录,里面真的有很多优秀的文章,欢迎大家的到来。 --- ';
|