ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

TensorOperatorWrapper

public class TensorOperatorWrapper

The adapter that makes a TensorOperator able to run with TensorImage.

Public Constructors

TensorOperatorWrapper(TensorOperator op)
Wraps a TensorOperator object as an ImageOperator, so that the TensorOperator could handle TensorImage objects by handling its underlying TensorBuffer.

Public Methods

TensorImage
int
getOutputImageHeight(int inputImageHeight, int inputImageWidth)
Computes the height of the expected output image when input image size is given.
int
getOutputImageWidth(int inputImageHeight, int inputImageWidth)
Computes the width of the expected output image when input image size is given.
PointF
inverseTransform(PointF point, int inputImageHeight, int inputImageWidth)
Transforms a point from coordinates system of the result image back to the one of the input image.

Inherited Methods

Public Constructors

public TensorOperatorWrapper (TensorOperator op)

Wraps a TensorOperator object as an ImageOperator, so that the TensorOperator could handle TensorImage objects by handling its underlying TensorBuffer.

Requirement: The op should not change coordinate system when applied on an image.

Parameters
op The created operator.

Public Methods

public TensorImage apply (TensorImage image)

Parameters
image

public int getOutputImageHeight (int inputImageHeight, int inputImageWidth)

Computes the height of the expected output image when input image size is given.

Parameters
inputImageHeight
inputImageWidth

public int getOutputImageWidth (int inputImageHeight, int inputImageWidth)

Computes the width of the expected output image when input image size is given.

Parameters
inputImageHeight
inputImageWidth

public PointF inverseTransform (PointF point, int inputImageHeight, int inputImageWidth)

Transforms a point from coordinates system of the result image back to the one of the input image.

Parameters
point the point from the result coordinates system.
inputImageHeight the height of input image.
inputImageWidth the width of input image.
Returns
  • the point with the coordinates from the coordinates system of the input image.