blueoil.datasets.div2k Module Contents Classes


Dataset base class

class blueoil.datasets.div2k.Div2k(subset='train', batch_size=10, augmentor=None, pre_processor=None, data_format='NHWC', seed=None, **kwargs)

Bases: blueoil.datasets.base.Base

Dataset base class

classes = []
num_classes = 0
extend_dir = DIV2K
available_subsets = ['train', 'validation']
property files(self)
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.