zokyo.augmentation.augment module

class zokyo.augmentation.augment.AbstractBuilder

Bases: abc.ABC

abstract classmethod get_keras_generator()

abstract classmethod process_and_generate()

abstract classmethod process_and_save()

class zokyo.augmentation.augment.Builder(config_json='config.json')

Bases: zokyo.augmentation.augment.AbstractBuilder

Builder class to create augmentor Pipeline object through a config file

TODO : Parallelize saving images to disk

TODO : Parallelize operations from data pipeline

get_builder_logger()

Method to return builder’s logger.

get_keras_generator(batch_size=None, internal_batch=None, input_func=None, output_func=None, task='classification')

Method to return a Keras generator. If internal batch is not set, batch size value is used

process_and_generate(batch_size=None, internal_batch=None, infinite_generator=False)

Process the images and yields the results in batches. NOTE : On batch ingestion mode if both internal batch and output batch size is given, sample number cannot be achieved.

process_and_save(batch_save_size=None, internal_batch_size=None)

Process the files and save to disk