快捷方式

VisionDataset

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

用于创建与 torchvision 兼容的数据集的基类。必须重写 __getitem____len__ 方法。

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

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

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

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

注意

transformstransformtarget_transform 的组合是互斥的。

使用 VisionDataset 的示例

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

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

如何使用 CutMix 和 MixUp

如何使用 CutMix 和 MixUp

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源