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: 端到端目標檢測/分割示例