trw.transforms.transforms_resize

Module Contents

Classes

TransformResize

Resize a tensor to a fixed size

Functions

_transform_resize(feature_names, batch, size, mode)

trw.transforms.transforms_resize._transform_resize(feature_names, batch, size, mode)

class trw.transforms.transforms_resize.TransformResize(size: trw.basic_typing.ShapeX, criteria_fn: Optional[trw.transforms.transforms.CriteriaFn] = None, mode='linear')

Bases: trw.transforms.transforms.TransformBatchWithCriteria

Resize a tensor to a fixed size