Skip to content
Keras+tf1.x training management utilities (Py2/3)
Python Jupyter Notebook
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
tests
.gitignore
LICENSE
README.md
__init__.py
applications.py
generators.py
generic.py
image_utils.py
model_helper.py
tensor_ops.py

README.md

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:

  • 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.

You can’t perform that action at this time.