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 (string, optional) – 資料集的根目錄。僅用於 __repr__。

• transforms (callable, optional) – 一個函式/轉換，它接收影像和標籤，並返回兩者的轉換版本。

• transform (callable, optional) – 一個函式/變換，接受 PIL 影像並返回變換後的版本。例如，transforms.RandomCrop

• target_transform (可呼叫物件, 可選) – 一個函式/變換，接受目標並對其進行變換。

注意

transforms 和 transform 與 target_transform 的組合是互斥的。

使用 VisionDataset 的示例

如何使用 CutMix 和 MixUp

如何使用 CutMix 和 MixUp

變換 v2：端到端目標檢測/分割示例

變換 v2：端到端目標檢測/分割示例