tensorbay.opendataset.BSTLD.loader

tensorbay.opendataset.BSTLD.loader.BSTLD(path: str) tensorbay.dataset.dataset.Dataset[source]

Dataloader of the BSTLD dataset.

The file structure should be like:

<path>
    rgb/
        additional/
            2015-10-05-10-52-01_bag/
                <image_name>.jpg
                ...
            ...
        test/
            <image_name>.jpg
            ...
        train/
            2015-05-29-15-29-39_arastradero_traffic_light_loop_bag/
                <image_name>.jpg
                ...
            ...
    test.yaml
    train.yaml
    additional_train.yaml
Parameters

path – The root directory of the dataset.

Raises

ModuleImportError – When the module “yaml” can not be found.

Returns

Loaded Dataset instance.