nntoolbox.vision.transforms.batch module

Transform input by batch

class nntoolbox.vision.transforms.batch.BatchCompose(transforms: List)[source]

Bases: object

class nntoolbox.vision.transforms.batch.BatchHorizontalFlip[source]

Bases: nntoolbox.vision.transforms.batch.BatchLambdaTransform

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

Bases: object

class nntoolbox.vision.transforms.batch.BatchRotation180[source]

Bases: nntoolbox.vision.transforms.batch.BatchLambdaTransform

class nntoolbox.vision.transforms.batch.BatchRotation270[source]

Bases: nntoolbox.vision.transforms.batch.BatchLambdaTransform

class nntoolbox.vision.transforms.batch.BatchRotation90[source]

Bases: nntoolbox.vision.transforms.batch.BatchLambdaTransform

class nntoolbox.vision.transforms.batch.BatchVerticalFlip[source]

Bases: nntoolbox.vision.transforms.batch.BatchLambdaTransform

class nntoolbox.vision.transforms.batch.Identity[source]

Bases: object