pytorch 从Dataset类中获取数据
PyTorch Dataset 获取数据 类中
2023-09-14 09:11:21 时间
转自:https://www.jianshu.com/p/4818a1a4b5bd
1.介绍
Dataset类是为torch.utils.data.DataLoader做准备,支持两种类型的访问
* map-style datasets #__getitem__()
* iterable-style datasets #__iter__()
(1) print("trainDataset 的类型:", type(trainDataset)) >>> trainDataset 的类型: <class 'torchvision.datasets.mnist.MNIST'> (2) print("trainDataset 的长度:", len(trainDataset)) >>> trainDataset 的长度: 60000 (3) print("trainDataset[0] 的类型:", type(trainDataset[0])) print("trainDataset[0] 的长度:", len(trainDataset[0])) >>> trainDataset[0] 的类型: <class 'tuple'> trainDataset[0] 的长度: 2 (4) print("trainDataset[0][0] 的类型:", type(trainDataset[0][0])) print("trainDataset[0][0] 的形状:", trainDataset[0][0].shape) >>> trainDataset[0][0] 的类型: <class 'torch.Tensor'> trainDataset[0][0] 的形状: torch.Size([1, 28, 28]) (5) print("trainDataset[0][1] 的类型:", type(trainDataset[0][1])) print("trainDataset[0][1] :", trainDataset[0][1]) >>> trainDataset[0][1] 的类型: <class 'int'> trainDataset[0][1] : 5
从上述代码可以看到,能够通过一些方法去访问。
相关文章
- 【PyTorch】教程:torch.nn.ReLU
- 【PyTorch】教程:torch.nn.Sequential
- 【PyTorch】教程:torchvision 目标检测微调
- DL之IDE:深度学习环境安装之pytorch/torchvision版本和CUDA版本匹配(对应版本安装)、利用Pytorch查看自己电脑上CUDA版本及其相关信息之详细攻略
- mobileNetV1网络解析,以及实现(pytorch)
- Pytorch实现ECA
- 总结几个好用的CNN模块(Pytorch)
- 【pytorch】torch.unsqueeze() 和 torch.squeeze()==>扩充维度和降维
- pytorch在windows10上安装使用
- Pytorch实现Warm up + Cosine Anneal LR
- Pytorch--自己动手实现模型剪枝
- 【Pytorch】解决cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
- 【Pytorch】查看GPU是否可用
- pytorch nn.parameter反向传播问题
- 【Pytorch Lighting】第 1 章:PyTorch Lightning adventure
- 基于深度学习的三维重建(二):pytorch的简单操作及DataLoader、Dataset类简介