nntoolbox.vision.transforms.batch module

Transform input by batch

class nntoolbox.vision.transforms.batch.BatchCompose(transforms: List)

Bases: object

class nntoolbox.vision.transforms.batch.BatchHorizontalFlip

Bases: nntoolbox.vision.transforms.batch.BatchLambdaTransform

class nntoolbox.vision.transforms.batch.BatchLambdaTransform(fn: Callable[[torch.Tensor], torch.Tensor])

Bases: object

class nntoolbox.vision.transforms.batch.BatchRotation180

Bases: nntoolbox.vision.transforms.batch.BatchLambdaTransform

class nntoolbox.vision.transforms.batch.BatchRotation270

Bases: nntoolbox.vision.transforms.batch.BatchLambdaTransform

class nntoolbox.vision.transforms.batch.BatchRotation90

Bases: nntoolbox.vision.transforms.batch.BatchLambdaTransform

class nntoolbox.vision.transforms.batch.BatchVerticalFlip

Bases: nntoolbox.vision.transforms.batch.BatchLambdaTransform

class nntoolbox.vision.transforms.batch.Identity

Bases: object