blueoil.datasets.widerface Module Contents Classes


Dataset base class

class blueoil.datasets.widerface.WiderFace(subset='train', max_boxes=3, *args, **kwargs)

Bases: blueoil.datasets.base.ObjectDetectionBase

Dataset base class

classes = ['face']
available_subsets = ['train', 'validation']
extend_dir = WIDER_FACE
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.

__getitem__(self, i)

Returns the i-th item of the dataset.


returns the number of items in the dataset.