trw.callbacks.callback_reporting_augmentations

Module Contents

Classes

CallbackReportingAugmentations

Export sample augmentations.

Attributes

logger

trw.callbacks.callback_reporting_augmentations.logger

class trw.callbacks.callback_reporting_augmentations.CallbackReportingAugmentations(nb_samples=10, nb_augmentation=5, table_name='augmentations', split_name=None, uid_name='sample_uid')

Bases: trw.callbacks.callback.Callback

Export sample augmentations.

Augmentation are detected using the uid_name of a sample. Samples with the same uid over several epochs

first_epoch(self, options)

create_or_recreate_table(self, options)

__call__(self, options, history, model, losses, outputs, datasets, datasets_infos, callbacks_per_batch, **kwargs)