捷徑

INaturalist

class torchvision.datasets.INaturalist(root: Union[str, Path], version: str = '2021_train', target_type: Union[List[str], str] = 'full', transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, download: bool = False)[來源]

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: 超級分類,例如 “Amphibians”(兩棲動物)

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

  • transform (callable, 選擇性) – 一個函數/轉換,它接受一個 PIL 圖像並返回一個轉換後的版本。例如,transforms.RandomCrop

  • target_transform (callable, 選擇性) – 一個函數/轉換,它接受目標並轉換它。

  • download (bool, 選擇性) – 如果為真,則從網際網路下載資料集並將其放置在根目錄中。如果已下載資料集,則不會再次下載。

__getitem__(index: int) Tuple[Any, Any][source]
參數:

index (int) – 索引

傳回值:

(image, target),其中目標的類型由 target_type 指定。

傳回類型:

tuple

category_name(category_type: str, category_id: int) str[source]
參數:
  • category_type (str) – 可選值包括 “full”、“kingdom”、“phylum”、“class”、“order”、“family”、“genus” 或 “super”

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

傳回值:

類別的名稱

文件

取得 PyTorch 的全面開發者文件

檢視文件

教學

取得適合初學者和進階開發人員的深入教學

檢視教學

資源

尋找開發資源並獲得您的問題解答

檢視資源