megengine.data#

>>> import megengine.data as data
>>> import megengine.data.transform as T

DataLoader#

DataLoader

Provides a convenient way to iterate on a given dataset.

Dataset#

Dataset

所有映射式(map-style)数据集的抽象基类

ArrayDataset

适用于 numpy ndarray 数据的 Dataset 类。

StreamDataset

An abstract class for stream data.

Vision Dataset#

VisionDataset

ImageFolder

ImageFolder 是一个类,可有组织地从文件夹中加载图像数据和标签。

MNIST

MNIST 数据集。

CIFAR10

适用于 CIFAR10 元数据的 Dataset .

CIFAR100

适用于 CIFAR100 元数据的 Dataset .

PascalVOC

Pascal VOC 数据集.

COCO

MS COCO 数据集.

Cityscapes

Cityscapes 数据集.

Objects365

Objects365 数据集.

ImageNet

从原始文件或文件夹中加载 ImageNet 数据集。

Sampler#

Sampler

所有采样器的抽象基类。

MapSampler

字典数据集的采样器。

StreamSampler

从流数据集中采样。

SequentialSampler

按序列顺序对元素进行采样。

RandomSampler

不放回(不重复选取)地随机采样。

ReplacementSampler

有放回(可重复选取)地进行随机采样。

Infinite

无限采样装饰器,可以包装普通的采样器。

Transform#

Transform

在子类中重写 apply 方法

PseudoTransform

Vision Transform#

VisionTransform

计算机视觉(vision)中所有transform(数据增广)类函数的基类。

Compose

将几个转换组合在一起。

TorchTransformCompose

torchvision 中变换的 Compose 类,仅支持 PIL 图像,但不支持某些torchvision 中张量上的变换,比如 Normalize 和 ToTensor.

ToMode

将输入数据改变为指定格式。

Pad

填充输入数据。

Resize

调整输入数据大小。

ShortestEdgeResize

使用指定的短边调整输入数据的大小。

RandomResize

随机调整输入数据的尺寸(size)。

RandomCrop

随机裁剪输入数据。

RandomResizedCrop

将输入数据按随机尺寸和纵横比进行裁剪。

CenterCrop

在中心处裁剪给定的输入数据。

RandomHorizontalFlip

按给定的概率,随机水平翻转输入数据。

RandomVerticalFlip

按给定概率,随机垂直翻转输入数据。

Normalize

用均值和标准差将输入数据正则化。

GaussianNoise

向输入数据添加随机高斯噪声。

BrightnessTransform

调整输入数据的亮度。

SaturationTransform

调整输入数据的饱和度。

ContrastTransform

调整输入数据的对比度。

HueTransform

调整输入数据的色调。

ColorJitter

随机改变图像的亮度、对比度、饱和度和色调。

Lighting

将 AlexNet 风格的 "lighting" 增广应用于输入数据。

Collator#

Collator

Used for merging a list of samples to form a mini-batch of Tensor(s).