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 (string, optional) – 数据集的根目录。仅用于 `__repr__`。
transforms (callable, optional) – 一个函数/变换，接收图像和标签，并返回两者的变换后的版本。
transform (callable, optional) – 一个函数/变换，接收 PIL 图像并返回变换后的版本。例如，`transforms.RandomCrop`
target_transform (可调用对象, 可选) – 一个函数/变换，接受目标并对其进行变换。
注意
`transforms` 和 `transform` 与 `target_transform` 的组合是互斥的。
使用 `VisionDataset` 的示例
