Food101¶
- class torchvision.datasets.Food101(root: ~typing.Union[str, ~pathlib.Path], split: str = 'train', transform: ~typing.Optional[~typing.Callable] = None, target_transform: ~typing.Optional[~typing.Callable] = None, download: bool = False, loader: ~typing.Callable[[~typing.Union[str, ~pathlib.Path]], ~typing.Any] = <function default_loader>)[源代码]¶
-
Food-101 是一个具有挑战性的数据集,包含 101 种食物类别,共 101,000 张图片。对于每个类别,提供了 250 张经过手动审查的测试图片以及 750 张训练图片。特意地,训练图片没有进行清理,因此仍然包含一些噪声。这些噪声主要表现为鲜艳的颜色,有时甚至是错误的标签。所有图片都被缩放到最大边长为 512 像素。
- 参数:
root (str 或
pathlib.Path
) – 数据集的根目录。split (string, optional) – 数据集分割,支持
"train"
(默认) 和"test"
。transform (callable, optional) – 一个函数/变换,接收 PIL 图像或 torch.Tensor(取决于给定的加载器),并返回一个变换后的版本。例如,
transforms.RandomCrop
target_transform (可调用对象, 可选) – 一个函数/变换,接受目标并对其进行变换。
download (bool, optional) – 如果为 True,则从互联网下载数据集并将其放入根目录。如果数据集已下载,则不会再次下载。默认为 False。
loader (callable, optional) – 一个加载给定路径的图像的函数。默认情况下,它使用 PIL 作为图像加载器,但用户也可以传入
torchvision.io.decode_image
以直接将图像数据解码为张量。
- 特殊成员: