快捷方式

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)[源代码]

KITTI 数据集。

它对应于用于目标检测的“对象左侧彩色图像”数据集。

参数:
  • root (str 或 pathlib.Path) –

    图像下载到的根目录。如果 download=False,则预期以下文件夹结构

    <root>
        └── Kitti
            └─ raw
                ├── training
                |   ├── image_2
                |   └── label_2
                └── testing
                    └── image_2
    

  • train (bool, 可选) – 如果为真,则使用 train 分割,否则使用 test 分割。默认为 train

  • transform (callable, 可选) – 一个函数/转换,它接收一个 PIL 图像并返回一个转换后的版本。例如,transforms.PILToTensor

  • target_transform (callable, 可选) – 一个函数/转换,它接收目标并对其进行转换。

  • transforms (callable, 可选) – 一个函数/转换,它接收输入样本及其目标作为输入并返回一个转换后的版本。

  • download (bool, 可选) – 如果为真,则从互联网下载数据集并将其放入根目录。如果数据集已下载,则不会再次下载。

特殊成员:

__getitem__(index: int) Tuple[Any, Any][源代码]

获取给定索引处的项目。

参数:

index (int) – 索引

返回值:

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

  • type: str

  • truncated: float

  • occluded: int

  • alpha: float

  • bbox: float[4]

  • dimensions: float[3]

  • locations: float[3]

  • rotation_y: float

返回类型:

tuple

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得解答

查看资源