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 寵物資料集.

參數:

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

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

• target_types (string, string 序列, optional) –

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

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

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

  • segmentation (PIL 圖片): 圖片的分割 trimap。

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

• transform (callable, optional) – 一個函數/轉換，它接收 PIL 圖片並返回轉換後的版本。 例如，transforms.RandomCrop。

• target_transform (callable, optional) – 一個函數/轉換，它接收目標並轉換它。

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

特殊成員:

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

參數:

index (int) – 索引

返回值:

樣本和元數據，可以選擇由各自的轉換進行轉換。

返回類型:

(Any)