【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")

 

 

相关内容

热门资讯

一分钟内幕!科乐吉林麻将系统发... 一分钟内幕!科乐吉林麻将系统发牌规律,福建大玩家确实真的是有挂,技巧教程(有挂ai代打);所有人都在...
一分钟揭秘!微扑克辅助软件(透... 一分钟揭秘!微扑克辅助软件(透视辅助)确实是有挂(2024已更新)(哔哩哔哩);1、用户打开应用后不...
五分钟发现!广东雀神麻雀怎么赢... 五分钟发现!广东雀神麻雀怎么赢,朋朋棋牌都是是真的有挂,高科技教程(有挂方法)1、广东雀神麻雀怎么赢...
每日必看!人皇大厅吗(透明挂)... 每日必看!人皇大厅吗(透明挂)好像存在有挂(2026已更新)(哔哩哔哩);人皇大厅吗辅助器中分为三种...
重大科普!新华棋牌有挂吗(透视... 重大科普!新华棋牌有挂吗(透视)一直是有挂(2021已更新)(哔哩哔哩)1、完成新华棋牌有挂吗的残局...
二分钟内幕!微信小程序途游辅助... 二分钟内幕!微信小程序途游辅助器,掌中乐游戏中心其实存在有挂,微扑克教程(有挂规律)二分钟内幕!微信...
科技揭秘!jj斗地主系统控牌吗... 科技揭秘!jj斗地主系统控牌吗(透视)本来真的是有挂(2025已更新)(哔哩哔哩)1、科技揭秘!jj...
1分钟普及!哈灵麻将攻略小,微... 1分钟普及!哈灵麻将攻略小,微信小程序十三张好像存在有挂,规律教程(有挂技巧)哈灵麻将攻略小是一种具...
9分钟教程!科乐麻将有挂吗,传... 9分钟教程!科乐麻将有挂吗,传送屋高防版辅助(总是存在有挂)1、完成传送屋高防版辅助透视辅助安装,帮...
每日必看教程!兴动游戏辅助器下... 每日必看教程!兴动游戏辅助器下载(辅助)真是真的有挂(2025已更新)(哔哩哔哩)1、打开软件启动之...