视觉数据集

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__方法。

Parameters:

• root (string, optional) – 数据集的根目录。仅用于 __repr__。

• transforms (callable, optional) – 一个函数/转换，接收一个图像和一个标签，并返回两者的转换版本。

• transform (callable, optional) – 一个函数/变换，接收一个PIL图像并返回一个变换后的版本。例如，transforms.RandomCrop

• target_transform (callable, optional) – 一个函数/转换，它接收目标并对其进行转换。

注意

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

使用 VisionDataset 的示例：

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

Transforms v2: End-to-end object detection/segmentation example

如何使用CutMix和MixUp

How to use CutMix and MixUp