Skip to content
/ ku Public

Training management utilities for TensorFlow Keras.

License

Notifications You must be signed in to change notification settings

subpic/ku

Repository files navigation

Keras Utilities (KU)

The project contains utilities for image assessment development with Keras/Tensorflow, including utilities for model training, custom generators, image management and augmentation. This is an extension of kutils.

Overview

Some of the key components of each file:

model_helper.py:

  • ModelHelper: Wrapper class that simplifies default usage of Keras for regression models.

generators.py:

  • DataGeneratorDisk, DataGeneratorHDF5: Keras generators for on-disk images, and HDF5 stored features/images

image_utils.py:

  • various utility functions for manipulating images (read, write to HDF5, batch resize, view batch)

image_augmenter.py:

  • ImageAugmenter: Create custom image augmentation functions for training Keras models.

generic.py:

  • H5Helper: Manage named data sets in HDF5 files, for us in Keras generators.
  • ShortNameBuilder: Utility for building short (file) names that contain multiple parameters.

applications.py:

  • model_inception_multigap, model_inceptionresnet_multigap: Model definitions for extracting MLSP narrow features
  • model_inception_pooled, model_inceptionresnet_pooled: Model definitions for extracting MLSP wide features

You can find more information in the docstrings.