×
PyTorch教程PyTorch简介PyTorch安装神经网络数学构建模块PyTorch神经网络基础机器学习与深度学习实现神经网络神经网络到功能块PyTorch术语PyTorch加载数据PyTorch线性回归PyTorch卷积神经网络PyTorch递归神经网络PyTorch数据集PyTorch Convents简介PyTorch从Scratch训练ConventPyTorch Convents特征提取PyTorch Convents可视化PyTorch Convent进行序列处理PyTorch单词嵌入PyTorch递归神经网络

PyTorch加载数据


PyTorch包含一个名为torchvision的包,用于加载和准备数据集。它包括两个基本功能,即DatasetDataLoader,它们有助于数据集的转换和加载。

数据集

数据集用于从给定数据集读取和转换数据点。实现的基本语法如下所述 -

trainset = torchvision.datasets.CIFAR10(root = './data', train = True,
   download = True, transform = transform)

DataLoader用于随机播放和批量处理数据。它可用于与多处理工作程序并行加载数据。

trainloader = torch.utils.data.DataLoader(trainset, batch_size = 4,
   shuffle = True, num_workers = 2)

示例:加载CSV文件

使用Python包Panda来加载csv文件。原始文件具有以下格式:(图像名称,68个标记 - 每个标记具有xy坐标)。

landmarks_frame = pd.read_csv('faces/face_landmarks.csv')

n = 65
img_name = landmarks_frame.iloc[n, 0]
landmarks = landmarks_frame.iloc[n, 1:].as_matrix()
landmarks = landmarks.astype('float').reshape(-1, 2)

分类导航

关注微信下载离线手册

bootwiki移动版 bootwiki
(群号:472910771)