waymo_open_dataset

Открытый набор данных Waymo состоит из данных датчиков высокого разрешения, собранных беспилотными автомобилями Waymo в самых разных условиях. Эти данные лицензированы для некоммерческого использования.

FeaturesDict({
    'camera_FRONT': FeaturesDict({
        'image': Image(shape=(1280, 1920, 3), dtype=uint8),
        'labels': Sequence({
            'bbox': BBoxFeature(shape=(4,), dtype=float32),
            'type': ClassLabel(shape=(), dtype=int64, num_classes=5),
        }),
    }),
    'camera_FRONT_LEFT': FeaturesDict({
        'image': Image(shape=(1280, 1920, 3), dtype=uint8),
        'labels': Sequence({
            'bbox': BBoxFeature(shape=(4,), dtype=float32),
            'type': ClassLabel(shape=(), dtype=int64, num_classes=5),
        }),
    }),
    'camera_FRONT_RIGHT': FeaturesDict({
        'image': Image(shape=(1280, 1920, 3), dtype=uint8),
        'labels': Sequence({
            'bbox': BBoxFeature(shape=(4,), dtype=float32),
            'type': ClassLabel(shape=(), dtype=int64, num_classes=5),
        }),
    }),
    'camera_SIDE_LEFT': FeaturesDict({
        'image': Image(shape=(886, 1920, 3), dtype=uint8),
        'labels': Sequence({
            'bbox': BBoxFeature(shape=(4,), dtype=float32),
            'type': ClassLabel(shape=(), dtype=int64, num_classes=5),
        }),
    }),
    'camera_SIDE_RIGHT': FeaturesDict({
        'image': Image(shape=(886, 1920, 3), dtype=uint8),
        'labels': Sequence({
            'bbox': BBoxFeature(shape=(4,), dtype=float32),
            'type': ClassLabel(shape=(), dtype=int64, num_classes=5),
        }),
    }),
    'context': FeaturesDict({
        'name': Text(shape=(), dtype=string),
    }),
    'timestamp_micros': int64,
})
  • Документация по функциям :
Особенность Учебный класс Форма Dтип Описание
ОсобенностиDict
camera_FRONT ОсобенностиDict
camera_FRONT/изображение Изображение (1280, 1920, 3) uint8
camera_FRONT/метки Последовательность
camera_FRONT/метки/bbox BBoxFeature (4,) поплавок32
camera_FRONT/метки/тип Метка класса int64
камера_FRONT_LEFT ОсобенностиDict
камера_FRONT_LEFT/изображение Изображение (1280, 1920, 3) uint8
camera_FRONT_LEFT/метки Последовательность
camera_FRONT_LEFT/этикетки/bbox BBoxFeature (4,) поплавок32
camera_FRONT_LEFT/метки/тип Метка класса int64
камера_FRONT_RIGHT ОсобенностиDict
camera_FRONT_RIGHT/изображение Изображение (1280, 1920, 3) uint8
camera_FRONT_RIGHT/метки Последовательность
camera_FRONT_RIGHT/labels/bbox BBoxFeature (4,) поплавок32
camera_FRONT_RIGHT/метки/тип Метка класса int64
камера_SIDE_LEFT ОсобенностиDict
camera_SIDE_LEFT/изображение Изображение (886, 1920, 3) uint8
camera_SIDE_LEFT/метки Последовательность
camera_SIDE_LEFT/этикетки/bbox BBoxFeature (4,) поплавок32
camera_SIDE_LEFT/метки/тип Метка класса int64
камера_SIDE_RIGHT ОсобенностиDict
camera_SIDE_RIGHT/изображение Изображение (886, 1920, 3) uint8
camera_SIDE_RIGHT/метки Последовательность
camera_SIDE_RIGHT/labels/bbox BBoxFeature (4,) поплавок32
camera_SIDE_RIGHT/метки/тип Метка класса int64
контекст ОсобенностиDict
контекст/имя Текст нить
timestamp_micros Тензор int64
@InProceedings{Sun_2020_CVPR,
author = {Sun, Pei and Kretzschmar, Henrik and Dotiwalla, Xerxes and Chouard, Aurelien and Patnaik, Vijaysai and Tsui, Paul and Guo, James and Zhou, Yin and Chai, Yuning and Caine, Benjamin and Vasudevan, Vijay and Han, Wei and Ngiam, Jiquan and Zhao, Hang and Timofeev, Aleksei and Ettinger, Scott and Krivokon, Maxim and Gao, Amy and Joshi, Aditya and Zhang, Yu and Shlens, Jonathon and Chen, Zhifeng and Anguelov, Dragomir},
title = {Scalability in Perception for Autonomous Driving: Waymo Open Dataset},
booktitle = {The IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2020}
}

waymo_open_dataset/v1.2 (конфигурация по умолчанию)

  • Описание конфигурации : Waymo Open Dataset v1.2

  • Размер набора данных : 336.62 GiB

  • Сплиты :

Расколоть Примеры
'train' 158 081
'validation' 39 987

waymo_open_dataset/v1.1

  • Описание конфигурации : Waymo Open Dataset v1.1

  • Размер набора данных : 336.62 GiB

  • Сплиты :

Расколоть Примеры
'train' 158 081
'validation' 39 987

waymo_open_dataset/v1.0

  • Описание конфигурации : Waymo Open Dataset v1.0 Этот набор данных также доступен в предварительно обработанном формате, что ускоряет его загрузку, если вы выберете правильный data_dir:
tfds.load('waymo_open_dataset/v1.0', data_dir='gs://waymo_open_dataset_v_1_0_0_individual_files/tensorflow_datasets')
  • Размер набора данных : 34.73 GiB

  • Сплиты :

Расколоть Примеры
'train' 14 884
'validation' 4954