VisionDataset

class torchvision.datasets.VisionDataset(root: Optional[Union[str, Path]] = None, transforms: Optional[Callable] = None, transform: Optional[Callable] = None, target_transform: Optional[Callable] = None)

用于创建兼容 torchvision 的数据集的基类。需要重写 __getitem__ 和 __len__ 方法。

参数：

• root (字符串, 可选) – 数据集的根目录。仅用于 __repr__。

• transforms (可调用对象, 可选) – 一个函数/变换，接收图像和标签并返回它们的变换后版本。

• transform (可调用对象, 可选) – 一个函数/变换，接收 PIL 图像并返回其变换后版本。例如，transforms.RandomCrop

• target_transform (可调用对象, 可选) – 一个函数/变换，接收目标并对其进行变换。

注意

transforms 和 transform 与 target_transform 的组合是互斥的。

使用 VisionDataset 的示例

如何使用 CutMix 和 MixUp

如何使用 CutMix 和 MixUp

Transforms v2: 端到端目标检测/分割示例

Transforms v2: 端到端目标检测/分割示例