快捷方式

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)[source]

iNaturalist 数据集。

参数:
  • root (str or 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 (可调用, 可选) – 一个函数/变换,它接收 PIL 图像并返回一个变换后的版本。例如,transforms.RandomCrop

  • target_transform (可调用, 可选) – 一个函数/变换,它接收目标并对其进行变换。

  • download (布尔值, 可选) – 如果为 True,则从互联网下载数据集并将其放在根目录中。如果数据集已下载,则不会再次下载。

__getitem__(index: int) Tuple[Any, Any][source]
参数:

index (整数) – 索引

返回值:

(image, target),其中 target 的类型由 target_type 指定。

返回值类型:

元组

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 的全面开发人员文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源