Package ai.djl.modality.cv.output

Class Mask

java.lang.Object

ai.djl.modality.cv.output.Rectangle

ai.djl.modality.cv.output.Mask

All Implemented Interfaces:

BoundingBox, Serializable

public class Mask
extends Rectangle

A mask with a probability for each pixel within a bounding rectangle.

This class is usually used to record the results of an Image Segmentation task.

See Also:

• Serialized Form

Constructor Summary

Constructors

Constructor	Description
Mask(double x, double y, double width, double height, float[][] dist)	Constructs a Mask with the given data.
Mask(double x, double y, double width, double height, float[][] dist, boolean fullImageMask)	Constructs a Mask with the given data.

Method Summary

Modifier and Type	Method	Description
float[][]	getProbDist()	Returns the probability for each pixel.
boolean	isFullImageMask()	Returns if the mask is for full image.

Methods inherited from class ai.djl.modality.cv.output.Rectangle

getBounds, getHeight, getIoU, getPath, getPoint, getWidth, getX, getY, nms, toString

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Details

Mask

public Mask(double x,
 double y,
 double width,
 double height,
 float[][] dist)

Constructs a Mask with the given data.

Parameters:

x - the left coordinate of the bounding rectangle

y - the top coordinate of the bounding rectangle

width - the width of the bounding rectangle

height - the height of the bounding rectangle

dist - the probability distribution for each pixel in the rectangle

Mask

public Mask(double x,
 double y,
 double width,
 double height,
 float[][] dist,
 boolean fullImageMask)

Constructs a Mask with the given data.

Parameters:

x - the left coordinate of the bounding rectangle

y - the top coordinate of the bounding rectangle

width - the width of the bounding rectangle

height - the height of the bounding rectangle

dist - the probability distribution for each pixel in the rectangle

fullImageMask - if the mask if for full image

Method Details

getProbDist

public float[][] getProbDist()

Returns the probability for each pixel.

Returns:

the probability for each pixel

isFullImageMask

public boolean isFullImageMask()

Returns if the mask is for full image.

Returns:

if the mask is for full image