nntoolbox.vision.transforms.imgaug module

Wrapper for imgaug transformation

class nntoolbox.vision.transforms.imgaug.ImgAugTransform(transform: imgaug.augmenters.meta.Augmenter)[source]

Bases: object

A thin wrapper around img aug transform

class nntoolbox.vision.transforms.imgaug.NumpyToTensor[source]

Bases: object

A simple class to transform a (batch) of numpy images to tensor

class nntoolbox.vision.transforms.imgaug.ToFloat[source]

Bases: object

A simple transform to convert image(s) from uint8 to float