OxfordIIITPet

class torchvision.datasets.OxfordIIITPet(root: Union[str, Path], split: str = 'trainval', target_types: Union[Sequence[str], str] = 'category', transforms: Optional[Callable] = None, transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, download: bool = False)

Oxford-IIIT Pet 資料集.

引數：

• root (str 或 pathlib.Path) – 資料集的根目錄。

• split (string, 可選) – 資料集劃分，支援 "trainval" (預設) 或 "test"。

• target_types (string, sequence of strings, 可選) –

  要使用的目標型別。可以是 category (預設) 或 segmentation。也可以是一個列表，輸出一個包含所有指定目標型別的元組。型別代表

  • category (int): 37種寵物類別之一的標籤。

  • binary-category (int): 貓或狗的二分類標籤。

  • segmentation (PIL image): 影像的分割三色圖。

  如果為空，將返回 None 作為目標。

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

• target_transform (callable, 可選) – 一個函式/變換，接收目標並對其進行變換。

• transforms (callable, 可選) – 一個函式/變換，將輸入樣本及其目標作為輸入，並返回變換後的版本。

• download (bool, 可選) – 如果為 True，則從網際網路下載資料集並將其放置到 root/oxford-iiit-pet 中。如果資料集已經下載，則不會再次下載。

特殊成員：

__getitem__(idx: int) → Tuple[Any, Any]

引數：

index (int) – 索引

返回：

樣本和元資料，可選地由相應的變換進行變換。

返回型別：

(Any)