PyTorch 数据集

PyTorch 数据集

在这一章中,我们将更多地关注 torchvision.datasets 及其各种类型。PyTorch包括以下数据集加载器:

MNIST

COCO(字幕和检测)

数据集包括以下两种主要类型的函数:

Transform - 一个接受图像并返回修改后的标准内容版本的函数。这些可以与转换一起组合使用。

Target_transform - 一个接受目标并进行转换的函数。例如,接收字幕字符串并返回单词索引的张量。

MNIST

以下是MNIST数据集的示例代码:

dset.MNIST(root, train = TRUE, transform = NONE,

target_transform = None, download = FALSE)

参数解释如下:

root − 数据集的根目录,其中存在处理后的数据。

train − True = 训练集,False = 测试集。

download − True = 从互联网下载数据集并放置在根目录下。

COCO

需要安装COCO API。以下示例用于展示使用PyTorch进行COCO数据集实现:

import torchvision.dataset as dset

import torchvision.transforms as transforms

cap = dset.CocoCaptions(root = ‘ dir where images are’,

annFile = ’json annotation file’,

transform = transforms.ToTensor())

print(‘Number of samples: ‘, len(cap))

print(target)

实现的输出如下 –

Number of samples: 82783

Image Size: (3L, 427L, 640L)

PyTorch 教程目录

PyTorch 简介

PyTorch 安装

PyTorch 神经网络的数学基本构建模块

PyTorch 神经网络基础

PyTorch 机器学习的通用工作流程

PyTorch 机器学习 vs. 深度学习

PyTorch 实现第一个神经网络

PyTorch 神经网络到功能块

PyTorch 术语

PyTorch 加载数据

PyTorch 线性回归

PyTorch 卷积神经网络

PyTorch 循环神经网络

PyTorch 数据集

PyTorch 卷积神经网络介绍

PyTorch 从头训练一个卷积神经网络

PyTorch 卷积神经网络中的特征提取

PyTorch 卷积神经网络的可视化

PyTorch 序列处理与卷积

PyTorch 词嵌入

PyTorch 递归神经网络

友情链接