kaishi.image.transforms.fix_rotation

Class definition for fixing image rotation.

Module Contents

class kaishi.image.transforms.fix_rotation.TransformFixRotation

Bases: kaishi.core.pipeline_component.PipelineComponent

Fix rotations of each image in a dataset given pre-determined labels (uses the default convnet for labels).

__call__(self, dataset)

Perform the transformation operation on an image dataset.

Parameters

dataset (kaishi.image.dataset.ImageDataset) – image dataset to perform operation on