8.1.1.3.1.2. blueoil.datasets.bdd100k

8.1.1.3.1.2.1. Module Contents

8.1.1.3.1.2.1.1. Classes

BDD100KObjectDetection

BDD100K Dataset for Object Detection (Car Camera)

BDD100KSegmentation

BDD100K Dataset for Segmentation

class blueoil.datasets.bdd100k.BDD100KObjectDetection(subset='train', max_boxes=100, *args, **kwargs)

Bases: blueoil.datasets.base.ObjectDetectionBase

BDD100K Dataset for Object Detection (Car Camera) https://github.com/ucbdrive/bdd-data

classes = ['bike', 'bus', 'car', 'motor', 'person', 'rider', 'traffic_light', 'traffic_sign', 'train', 'truck']
num_classes
available_subsets = ['train', 'validation']
extend_dir = bdd100k
property label_colors(self)
classmethod count_max_boxes(cls, base_path=None)

Count max boxes size over all subsets.

property num_max_boxes(self)

Return count max box size of available subsets.

property num_per_epoch(self)

Returns the number of datas in the data subset.

_init_files_and_annotations(self)
__getitem__(self, i)

Returns the i-th item of the dataset.

__len__(self)

returns the number of items in the dataset.

class blueoil.datasets.bdd100k.BDD100KSegmentation(batch_size=10, *args, **kwargs)

Bases: blueoil.datasets.base.SegmentationBase

BDD100K Dataset for Segmentation https://github.com/ucbdrive/bdd-data

available_subsets = ['train', 'validation', 'test']
extend_dir = seg
image_dir = images
label_dir = labels
classes = ['unlabeled', 'dynamic', 'ego_vehicle', 'ground', 'static', 'parking', 'rail track', 'road', 'sidewalk', 'bridge', 'building', 'fence', 'garage', 'guard rail', 'tunnel', 'wall', 'banner', 'billboard', 'lane divider', 'parking_sign', 'pole', 'polegroup', 'street_light', 'traffic_cone', 'traffic_device', 'traffic_light', 'traffic_sign', 'traffic_sign_frame', 'terrain', 'vegetation', 'sky', 'person', 'rider', 'bicycle', 'bus', 'car', 'caravan', 'motorcycle', 'trailer', 'train', 'truck']
num_classes
property label_colors(self)
files_and_annotations(self)
__getitem__(self, i)

Returns the i-th item of the dataset.

__len__(self)

returns the number of items in the dataset.

property num_per_epoch(self)

Returns the number of datas in the data subset.