ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

transform的使用2

2022-02-20 15:30:00  阅读:182  来源: 互联网

标签:img writer 0.5 transform transforms 使用 trans resize


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

img = Image.open("/Users/computer/Documents/Code/pytorchLearning/imgs/five.png")
trans_toTensor = transforms.ToTensor()
tran_img = trans_toTensor(img)
writer = SummaryWriter("logs")
writer.add_image("tran_img", tran_img)

# Normalize(归一化)
transforms.Normalize([0.5,0.5,0.5],[0.5,0.5,0.5])

# Resize (改变图形大小)
print(img.size)
trans_resize = transforms.Resize((512, 512))
img_resize = trans_resize(img)
img_resize = trans_toTensor(img_resize)
writer.add_image("img_resize", img_resize)
print(img_resize)

# Compose resize (组合)
tran_resize2 = transforms.Resize(512)
trans_compose = transforms.Compose([tran_resize2,trans_toTensor])
img_resize2 = trans_compose(img)
writer.add_image("img_resize2", img_resize2)

# RandomCrop (随机裁剪)
trans_random = transforms.RandomCrop((500, 1000))
trans_compose2 = transforms.Compose([trans_random, trans_toTensor])
for i in range(10):
    img_crop = trans_compose2(img)
    writer.add_image("RandomCrop", img_crop, i)

writer.close()

参考课程:https://www.bilibili.com/video/BV1hE411t7RN?p=12

标签:img,writer,0.5,transform,transforms,使用,trans,resize
来源: https://blog.csdn.net/weixin_43328054/article/details/123030915

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有