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,则从互联网下载数据集并将其放在根目录中。如果数据集已下载,则不会再次下载。