pytorch 预处理

裁剪，遮挡，数据增强：

import torch as t
import numpy as np
import torchvision as tv
import matplotlib.pyplot as plt
from torchvision import transforms
from torchtoolbox.transform import CutoutROOT = '../pytorch/cifar-10'
BATCH_SIZE=128train_transform = transforms.Compose([transforms.RandomCrop(32, padding=4),#每边填充4，把32^*32填充至40*40，再随机裁剪Cutout(0.5),#参数是遮挡的概率transforms.RandomHorizontalFlip(),transforms.ToTensor()
])

人脸识别增强:

self.transform = transforms.Compose([transforms.ToPILImage(),transforms.RandomHorizontalFlip(),transforms.ToTensor(),transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]),])

resize只能针对 pil image起作用，

如果输入时opencv 读取的图片numpy数组，则把resize去掉即可。

resize不支持opencv

pytorch 预处理相关推荐

目标检测之Faster-RCNN的pytorch代码详解(数据预处理篇)
首先贴上代码原作者的github:https://github.com/chenyuntc/simple-faster-rcnn-pytorch(非代码作者,博文只解释代码) 今天看完了simple- ...
英伟达DALI加速技巧：使数据预处理比原生PyTorch运算速度快4倍
关注上方"深度学习技术前沿",选择"星标公众号", 资源干货,第一时间送达! 你的数据处理影响整个训练速度,如果加上英伟达 DALI 库,处理速度比原生 PyT ...
PyTorch中的数据输入和预处理
文章目录 PyTorch中的数据输入和预处理数据载入类映射类型的数据集 torchvision工具包的使用可迭代类型的数据集总结 PyTorch中的数据输入和预处理数据载入类在使用PyTo ...
PyTorch基础（四）-----数据加载和预处理
前言之前已经简单讲述了PyTorch的Tensor.Autograd.torch.nn和torch.optim包,通过这些我们已经可以简单的搭建一个网络模型,但这是不够的,我们还需要大量的数据,众所 ...
pytorch dataset dataloader_PyTorch（五）——数据的加载和预处理
前言 PyTorch通过torch.utils.data对一般的常用数据进行封装,可以很容易地实现多线程数据预读和批量加载.torchvision已经预先实现了常用的图像数据集,包括CIFAR-10. ...
PyTorch 系列 | 数据加载和预处理教程
图片来源:Unsplash,作者:Damiano Baschiera 2019 年第 66 篇文章,总第 90 篇文章本文大约 8000 字,建议收藏阅读原题 | DATA LOADING AND ...
【Pytorch神经网络基础理论篇】 03 数据操作 + 数据预处理
1.数组样例 2.数据操作(代码实现与结果截图) #首先导入torch,虽然被称为pytorch,但是我们应该导入torch而不是pytorch import torch #张量表示一个数值组成的数组 ...
PyTorch框架学习六——图像预处理transforms（二）
PyTorch框架学习六--图像预处理transforms(二) (续)二.transforms的具体方法 4.图像变换 (1)尺寸变换:transforms.Resize() (2)标准化:tran ...
PyTorch框架学习五——图像预处理transforms（一）
PyTorch框架学习五--图像预处理transforms(一) 一.transforms运行机制二.transforms的具体方法 1.裁剪 (1)随机裁剪:transforms.RandomCr ...

pytorch 预处理

pytorch 预处理相关推荐

最新文章

热门文章