VOCDetection¶
- class torchvision.datasets.VOCDetection(root: Union[str, Path], year: str = '2012', image_set: str = 'train', download: bool = False, transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, transforms: Optional[Callable] = None)[source]¶
Pascal VOC 检测数据集。
- 参数:
root (
str
或pathlib.Path
) – VOC 数据集的根目录。year (字符串, 可选) – 数据集年份,支持年份
"2007"
到"2012"
。image_set (字符串, 可选) – 选择要使用的图像集,
"train"
、"trainval"
或"val"
。如果year=="2007"
,也可以是"test"
。download (布尔值, 可选) – 如果为真,则从互联网下载数据集并将其放入根目录。如果数据集已下载,则不会再次下载。(默认:VOC 的 20 个类的字母顺序索引)。
transform (可调用对象, 可选) – 一个函数/转换,它接收一个 PIL 图像并返回一个转换后的版本。例如,
transforms.RandomCrop
target_transform (可调用对象, 必需) – 一个函数/转换,它接收目标并将其转换。
transforms (可调用对象, 可选) – 一个函数/转换,它以输入样本及其目标作为输入,并返回一个转换后的版本。
- 特殊成员: