brails.processors.garage_detector.lib.efficientdet.dataset module

class brails.processors.garage_detector.lib.efficientdet.dataset.Augmenter

Bases: object

Convert ndarrays in sample to Tensors.

class brails.processors.garage_detector.lib.efficientdet.dataset.CocoDataset(root_dir, coco_dir, img_dir, set_dir='train2017', transform=None)

Bases: Dataset

load_annotations(image_index)

load_classes()

load_image(image_index)

class brails.processors.garage_detector.lib.efficientdet.dataset.Normalizer(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]): Bases: object

class brails.processors.garage_detector.lib.efficientdet.dataset.Resizer(img_size=512)

Bases: object

Convert ndarrays in sample to Tensors.

brails.processors.garage_detector.lib.efficientdet.dataset.collater(data)