Kitti¶

class torchvision.datasets.Kitti(root: Union[str, Path], train: bool = True, transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, transforms: Optional[Callable] = None, download: bool = False)[source]¶

KITTI 数据集。

它对应于“物体的左彩色图像”数据集，用于物体检测。

Parameters:

root (str or pathlib.Path) –

图像下载到的根目录。如果download=False，期望以下文件夹结构：

    └── Kitti
        └─ raw
            ├── training
            |   ├── image_2
            |   └── label_2
            └── testing
                └── image_2

train (bool, 可选) – 如果为真，则使用 train 分割，否则使用 test 分割。默认为 train。
transform (callable, optional) – 一个函数/变换，接收一个PIL图像并返回一个变换后的版本。例如，transforms.PILToTensor
target_transform (callable, optional) – 一个函数/转换，它接收目标并对其进行转换。
transforms (callable, optional) – 一个函数/转换，它接收输入样本及其目标作为输入，并返回转换后的版本。
下载 (bool, 可选) – 如果为真，则从互联网下载数据集并将其放入根目录。如果数据集已经下载，则不会再次下载。

Special-members:

__getitem__(index: int) → Tuple[Any, Any][source]¶

获取给定索引处的项目。

Parameters:

index (int) – 索引

Returns:

(image, target)，其中 target 是一个包含以下键的字典列表：

type: str
truncated: float
occluded: int
alpha: float
bbox: float[4]
dimensions: float[3]
locations: float[3]
rotation_y: float

Return type:

tuple