视觉数据集

class VisionDataset(images: Sequence[str | Path | Tensor], transforms: Sequence | None = None, root: Path | None = None)[source]

基础类：Dataset

一个带有数据增强的图像数据集。

注意

需要安装torchvision。

初始化数据集。

Parameters:

  • images (Sequence[str | Path | Tensor]) – 图像，可以是（相对）路径，也可以是预处理后的张量。

  • transforms (Sequence | None) – 应用于图像的一系列变换， 参见 torchvision.transforms

  • root (Path | None) – 图像的根目录