24、其他api函数以及两种自动增强的方法

24、针对tensor操作,随机选择矩形区域删除
torchvision.transforms.RandomErasing(p=0.5, scale=(0.02, 0.33), ratio=(0.3, 3.3), value=0, inplace=False)
25、针对tensor操作,归一化图片,给定均值和方差
torchvision.transforms.Normalize(mean, std, inplace=False)
26、给定tensor数据格式进行改变
torchvision.transforms.ConvertImageDtype(dtype: torch.dtype)
27、把numpy(H x W x C) or tensor( C x H x W) 转为PIL
torchvision.transforms.ToPILImage(mode=None)
1、将Tensor的每个元素乘以2552、将数据由Tensor转化成Uint8;
3、将Tensor转化成numpy的ndarray类型;
4、对ndarray对象做permute (1, 2, 0)的转置,因为pytorch的维度和图片维度不一样;
5、将ndarray对象转化成PILImage数据格式;
6、输出该PILImage数据(save后可以直接打开)。
28、把PIL or numpy 转为tensor
torchvision.transforms.ToTensor
1、将图片转化成内存中的存储格式;
2、将字节以流的形式输入,转化成Tensor类型;
3、对Tensor进行reshape;
4、对Tensor进行permute(2,0,1,因为pytorch的维度和图片维度不一样;
5、将Tensor中的每个元素除以2556、输出该Tensor数据。
27PIL(H x W x C)对象转为tensor(C x H x W)
torchvision.transforms.PILToTensor

#随机增强方法一
augmenter = transform.RandAugment()
img0=Image.open('lin-xiao-xun-000003.jpg')
imgs = [augmenter(img0) for _ in range(4)]
axs = plt.figure().subplots(1, 5)
axs[0].imshow(img0);axs[0].set_title('src');axs[0].axis('off')
axs[1].imshow(imgs[0]);axs[1].set_title('1');axs[1].axis('off')
axs[2].imshow(imgs[1]);axs[2].set_title('2');axs[2].axis('off')
axs[3].imshow(imgs[2]);axs[3].set_title('3');axs[3].axis('off')
axs[4].imshow(imgs[3]);axs[4].set_title('4');axs[4].axis('off')
plt.show()

图片:
在这里插入图片描述

#随机增强方法二

augmenter = transform.TrivialAugmentWide()
img0=Image.open('lin-xiao-xun-000003.jpg')
imgs = [augmenter(img0) for _ in range(4)]
axs = plt.figure().subplots(1, 5)
axs[0].imshow(img0);axs[0].set_title('src');axs[0].axis('off')
axs[1].imshow(imgs[0]);axs[1].set_title('1');axs[1].axis('off')
axs[2].imshow(imgs[1]);axs[2].set_title('2');axs[2].axis('off')
axs[3].imshow(imgs[2]);axs[3].set_title('3');axs[3].axis('off')
axs[4].imshow(imgs[3]);axs[4].set_title('4');axs[4].axis('off')
plt.show()

图片:
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值