pytorch学习笔记
猫养的仓鼠
这个作者很懒,什么都没留下…
展开
-
Pytorch数据读取(Dataset, DataLoader, DataLoaderIter)
Pytorch的数据读取主要包含三个类:Dataset DataLoader DataLoaderIter这三者大致是一个依次封装的关系: 1.被装进2., 2.被装进3.一. torch.utils.data.Dataset是一个抽象类, 自定义的Dataset需要继承它并且实现两个成员方法:__getitem__() __len__()第一个最为重要, 即每次怎...转载 2018-11-07 17:15:02 · 5678 阅读 · 0 评论 -
pytorch torchvision transform
一、对PIL.Image进行变换class torchvision.transforms.Compose(transforms)将多个 transform 组合起来使用。transforms: 由transform构成的列表. 例子:transforms.Compose([ transforms.CenterCrop(10), transforms.ToTe...转载 2018-11-07 18:59:41 · 277 阅读 · 0 评论