【Pytorch】数据集的加载和处理(二)
创始人
2025-01-10 23:06:00
0

【Pytorch】数据集的加载和处理(一) 

Pytorch中张量可以是一维、二维、三维或者更高维度的数据结构。一维张量类似于向量,二维张量类似于矩阵,三维张量类似一系列矩阵的堆叠。

目录

将张量包装为数据集

创建数据加载器

数据转换(图像转换)


将张量包装为数据集

导入MNIST训练数据集并提取数据和标签

import torch import torchvision from torchvision import datasets train_data=datasets.MNIST("./data",train=True,download=True) x_train, y_train=train_data.data,train_data.targets 

导入MNIST验证数据集并提取数据和标签

val_data=datasets.MNIST("./data", train=False, download=True) x_val,y_val=val_data.data, val_data.targets 

 使用 TensorDataset类将张量包装为数据集

from torch.utils.data import TensorDataset train_ds = TensorDataset(x_train, y_train) val_ds = TensorDataset(x_val, y_val)  for x,y in train_ds:     print(x.shape,y.item())     break

 

创建数据加载器

通过DataLoader从数据集创建数据加载器

from torch.utils.data import DataLoader train_dl = DataLoader(train_ds, batch_size=100) val_dl = DataLoader(val_ds, batch_size=100)  for xb,yb in train_dl:     print(xb.shape)     print(yb.shape)     break

 

数据转换(图像转换)

通过 transform 类进行简单的图像转换

导入库和训练数据集 

import torchvision import matplotlib.pyplot as plt from torchvision import datasets from torchvision import transforms train_data=datasets.MNIST("./data", train=True, download=True)

借助transform类定义旋转

data_transform = transforms.Compose ([     transforms.RandomHorizontalFlip(p=1),     transforms.RandomVerticalFlip(p=1),     transforms.ToTensor(), ])

对训练数据集中图像进行旋转并打印对比

img = train_data[5][0] img_tr=data_transform(img) img_tr_np=img_tr.numpy()  plt.subplot(1,2,1) plt.imshow(img,cmap="gray") plt.title("original") plt.subplot(1,2,2) plt.imshow(img_tr_np[0],cmap="gray"); plt.title("transformed 180")

 

 

相关内容

热门资讯

6分钟发现!掌电竞技真的能赢吗... 6分钟发现!掌电竞技真的能赢吗,菠萝德州好像真的有挂,揭秘教程(有挂脚本)1、用户打开应用后不用登录...
八分钟科普!天天微友辅助器通用... 八分钟科普!天天微友辅助器通用版,wEPOKE原来存在有挂,插件教程(有挂解说)1.天天微友辅助器通...
五分钟发现!优乐麻将挂,AAp... 五分钟发现!优乐麻将挂,AApOKER确实有挂,2025新版技巧(有挂工具)一、优乐麻将挂软件透明挂...
三分钟科普!中至辅助器免费版v... 三分钟科普!中至辅助器免费版v3.0,aapoker原来是有挂,德州论坛(有挂攻略)1、该软件可以轻...
9分钟攻略!边锋干瞪眼有外挂么... 9分钟攻略!边锋干瞪眼有外挂么,德扑ai总是真的有挂,安装教程(有挂ai代打)1、边锋干瞪眼有外挂么...
七分钟了解!闲娱棋牌有挂吗,w... 七分钟了解!闲娱棋牌有挂吗,wepower德州都是真的有挂,必备教程(有挂脚本);进入游戏-大厅左侧...
2分钟科普!雀神微信小程序怎么... 2分钟科普!雀神微信小程序怎么开挂,aaPOKER确实是真的有挂,玩家教你(有挂教学)1、点击下载安...
九分钟辅助!决战卡五星游戏辅助... 九分钟辅助!决战卡五星游戏辅助器,德州俱乐部果然有挂,详细教程(有挂揭秘)暗藏猫腻,小编详细说明决战...
8分钟辅助挂!财神十三张特殊牌... 8分钟辅助挂!财神十三张特殊牌外 挂,we-poker原来有挂,新2025版(有挂教学)1、每一步都...
7分钟辅助挂!台州宝宝游戏外 ... 7分钟辅助挂!台州宝宝游戏外 挂,鱼扑克app俱乐部切实是有挂,安装教程(有挂辅助挂)1、任何台州宝...