megengine.data

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

DataLoader

DataLoader

可用于在给定数据集上遍历并取得数据。

Dataset

Dataset	所有数据集类的抽象基类。
ArrayDataset	适用于 numpy ndarray 数据的 Dataset 类。
StreamDataset	适用于流式数据的抽象类。

Vision Dataset

VisionDataset
ImageFolder	ImageFolder 是一个类，可有组织地从文件夹中加载图像数据和标签。
MNIST	适用于 MNIST 元数据的 Dataset .
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).

上一页

megengine.set_log_level

下一页

megengine.data.DataLoader