tensorbay.opendataset.BDD100K.loader

This file defines the BDD100K dataloader and the BDD100K_10K dataloader.

tensorbay.opendataset.BDD100K.loader.BDD100K(path: str) → tensorbay.dataset.dataset.Dataset[source]

Dataloader of the BDD100K dataset.

The file structure should be like:

<path>
    bdd100k_images_100k/
        images/
            100k/
                test
                train
                val
        labels/
            det_20/
                det_train.json
                det_val.json
            lane/
                polygons/
                    lane_train.json
                    lane_val.json
            drivable/
                polygons/
                    drivable_train.json
                    drivable_val.json

Parameters: path – The root directory of the dataset.
Returns: Loaded Dataset instance.

tensorbay.opendataset.BDD100K.loader.BDD100K_10K(path: str) → tensorbay.dataset.dataset.Dataset[source]

Dataloader of the BDD100K_10K dataset.

The file structure should be like:

<path>
    bdd100k_images_10k/
        images/
            10k/
                test/
                    cabc30fc-e7726578.jpg
                    ...
                train/
                    0a0a0b1a-7c39d841.jpg
                    ...
                val/
                    b1c9c847-3bda4659.jpg
                    ...
        labels/
            pan_seg/
                polygons/
                    pan_seg_train.json
                    pan_seg_val.json
                bitmasks/
                    train/
                        0a0a0b1a-7c39d841.png
                        ...
                    val/
                        b1c9c847-3bda4659.png
                        ...
            sem_seg/
                masks/
                    train/
                        0a0a0b1a-7c39d841.png
                        ...
                    val/
                        b1c9c847-3bda4659.png
                        ...
            ins_seg/
                bitmasks/
                    train/
                        0a0a0b1a-7c39d841.png
                        ...
                    val/
                        b1c9c847-3bda4659.png
                        ...

Parameters: path – The root directory of the dataset.
Returns: Loaded Dataset instance.