INaturalist

類 torchvision.datasets.INaturalist(root: 聯合[str, Path], version: str = '2021_train', target_type: 聯合[列表[str], str] = 'full', transform: 可選的[可呼叫物件] = None, target_transform: 可選的[可呼叫物件] = None, download: 布林值 = False, loader: 可選的[可呼叫物件[[聯合[str, Path]], 任意型別]] = None)

iNaturalist 資料集。

引數:

• root (str 或 pathlib.Path) – 資料集根目錄，用於儲存影像檔案。此類不需要/不使用標註檔案。

• version (字串, 可選) – 要下載/使用的資料集版本。可選值包括 '2017', '2018', '2019', '2021_train', '2021_train_mini', '2021_valid'。預設值: 2021_train。

• target_type (字串 或 列表, 可選) –

  目標型別，對於 2021 版本，可選值包括:

  • full: 完整類別（物種）

  • kingdom: 例如，“Animalia”

  • phylum: 例如，“Arthropoda”

  • class: 例如，“Insecta”

  • order: 例如，“Coleoptera”

  • family: 例如，“Cleridae”

  • genus: 例如，“Trichodes”

  對於 2017-2019 版本，可選值包括:

  • full: 完整（數字）類別

  • super: 超級類別，例如，“兩棲動物”

  也可以是列表，輸出包含所有指定目標型別的元組。預設值為 full。

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

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

• download (布林值, 可選) – 如果為 True，則從網際網路下載資料集並將其放在根目錄中。如果資料集已下載，則不會再次下載。

• loader (可呼叫物件, 可選) – 一個根據路徑載入影像的函式。預設情況下，它使用 PIL 作為其影像載入器，但使用者也可以傳入 torchvision.io.decode_image，以便直接將影像資料解碼為張量。

__getitem__(index: int) → 元組[任意型別, 任意型別]

引數:

index (int) – 索引

返回:

(影像, 目標)，其中目標的型別由 target_type 指定。

返回型別:

元組

category_name(category_type: str, category_id: int) → str

引數:

• category_type (str) – “full”, “kingdom”, “phylum”, “class”, “order”, “family”, “genus” 或 “super” 之一

• category_id (int) – 此類別中的索引（類別 ID）

返回:

類別名稱