CelebA¶
- class torchvision.datasets.CelebA(root: Union[str, Path], split: str = 'train', target_type: Union[List[str], str] = 'attr', transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, download: bool = False)[source]¶
-
- 参数:
root (str or
pathlib.Path
) – 图片下载到的根目录。split (string) – {‘train’, ‘valid’, ‘test’, ‘all’} 之一。相应地选择数据集。
target_type (string 或 list, 可选) –
要使用的目标类型,
attr
、identity
、bbox
或landmarks
。也可以是列表,以输出包含所有指定目标类型的元组。目标表示attr
(张量形状=(40,) 类型=int): 属性的二进制 (0, 1) 标签identity
(int): 每个人的标签(具有相同身份的数据点是同一个人)bbox
(张量形状=(4,) 类型=int): 边界框 (x, y, 宽度, 高度)landmarks
(张量形状=(10,) 类型=int): 关键点 (左眼_x, 左眼_y, 右眼_x, 右眼_y, 鼻子_x, 鼻子_y, 左嘴_x, 左嘴_y, 右嘴_x, 右嘴_y)
默认值为
attr
。如果为空,则返回None
作为目标。transform (可调用, 可选) – 一个函数/变换,它接收 PIL 图像并返回转换后的版本。例如,
transforms.PILToTensor
target_transform (可调用, 可选) – 一个函数/变换,它接收目标并对其进行变换。
download (bool, 可选) –
如果为真,则从互联网下载数据集并将其放入根目录。如果数据集已下载,则不会再次下载。
警告
要下载数据集,需要 gdown。
- 特殊成员: