斯坦福汽车¶
- class torchvision.datasets.StanfordCars(root: Union[str, Path], split: str = 'train', transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, download: bool = False)[source]¶
斯坦福汽车数据集
Cars数据集包含16,185张图片,涵盖196类汽车。数据分为8,144张训练图片和8,041张测试图片,每类大致按50-50的比例分割。
原始URL是https://ai.stanford.edu/~jkrause/cars/car_dataset.html,但它已失效。
注意
这个类需要scipy来从.mat格式加载目标文件。
- Parameters:
root (str 或
pathlib.Path) – 数据集的根目录split (string, optional) – 数据集的分割,支持
"train"(默认)或"test"。transform (callable, optional) – 一个函数/变换,接收一个PIL图像并返回一个变换后的版本。例如,
transforms.RandomCroptarget_transform (callable, optional) – 一个函数/转换,它接收目标并对其进行转换。
下载 (bool, 可选) – 此参数存在是为了向后兼容,但它不会下载数据集,因为原始URL已不再可用。数据集似乎在Kaggle上可用,因此您可以尝试使用这些说明手动下载它。
- Special-members: