Skip to content

Index

Preprocessing Layers API

This module provides a variety of preprocessing/augmentation layers to build custom tf.data.Dataset pipelines. NSE provides layers for both 1D and 2D input data and doesnt assume 2D input data to be images. In addition, all layers inherit from BaseAugmentation and TFDataLayer. These two layers provide the following functionalities:

  • Dynamically set backend to TensorFlow for pipeline layers
  • Coerce input data to have batch dimension and converted to nested dictionary
  • Output data will revert to original format (e.g. no batch)
  • By supporting nested dictionary, it allows layers to manipulate labels
  • The layers map either sequentially or in parallel across the batch dimension

Classes:

Classes