waymo_open_dataset

  • Description :

L'ensemble de données ouvertes Waymo est composé de données de capteurs haute résolution collectées par les voitures autonomes Waymo dans une grande variété de conditions. Ces données sont sous licence pour un usage non commercial.

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,
})
  • Documentation des fonctionnalités :
Fonctionnalité Classe Forme Type D Description
FonctionnalitésDict
caméra_FRONT FonctionnalitésDict
caméra_FRONT/image Image (1280, 1920, 3) uint8
camera_FRONT/étiquettes Séquence
camera_FRONT/labels/bbox Fonctionnalité BBox (4,) flotteur32
camera_FRONT/étiquettes/type Étiquette de classe int64
caméra_FRONT_LEFT FonctionnalitésDict
caméra_FRONT_LEFT/image Image (1280, 1920, 3) uint8
camera_FRONT_LEFT/étiquettes Séquence
camera_FRONT_LEFT/labels/bbox Fonctionnalité BBox (4,) flotteur32
camera_FRONT_LEFT/étiquettes/type Étiquette de classe int64
caméra_FRONT_RIGHT FonctionnalitésDict
caméra_FRONT_RIGHT/image Image (1280, 1920, 3) uint8
camera_FRONT_RIGHT/étiquettes Séquence
camera_FRONT_RIGHT/labels/bbox Fonctionnalité BBox (4,) flotteur32
camera_FRONT_RIGHT/étiquettes/type Étiquette de classe int64
caméra_SIDE_LEFT FonctionnalitésDict
caméra_SIDE_LEFT/image Image (886, 1920, 3) uint8
camera_SIDE_LEFT/étiquettes Séquence
camera_SIDE_LEFT/labels/bbox Fonctionnalité BBox (4,) flotteur32
camera_SIDE_LEFT/étiquettes/type Étiquette de classe int64
caméra_SIDE_RIGHT FonctionnalitésDict
caméra_SIDE_RIGHT/image Image (886, 1920, 3) uint8
camera_SIDE_RIGHT/étiquettes Séquence
camera_SIDE_RIGHT/étiquettes/bbox Fonctionnalité BBox (4,) flotteur32
camera_SIDE_RIGHT/étiquettes/type Étiquette de classe int64
contexte FonctionnalitésDict
contexte/nom Texte chaîne
horodatage_micros Tenseur 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 (configuration par défaut)

  • Description de la configuration : Waymo Open Dataset v1.2

  • Taille de l'ensemble de données : 336.62 GiB

  • Divisions :

Diviser Exemples
'train' 158 081
'validation' 39 987

waymo_open_dataset/v1.1

  • Description de la configuration : Waymo Open Dataset v1.1

  • Taille de l'ensemble de données : 336.62 GiB

  • Divisions :

Diviser Exemples
'train' 158 081
'validation' 39 987

waymo_open_dataset/v1.0

  • Description de la configuration : Waymo Open Dataset v1.0 Cet ensemble de données est également disponible au format prétraité, ce qui le rend plus rapide à charger, si vous sélectionnez le bon data_dir :
tfds.load('waymo_open_dataset/v1.0', data_dir='gs://waymo_open_dataset_v_1_0_0_individual_files/tensorflow_datasets')
  • Taille de l'ensemble de données : 34.73 GiB

  • Divisions :

Diviser Exemples
'train' 14 884
'validation' 4 954