ai.djl.modality.cv

Class Mask

java.lang.Object

ai.djl.modality.cv.Rectangle

ai.djl.modality.cv.Mask

All Implemented Interfaces:

BoundingBox

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.

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
void	draw(java.awt.Graphics2D g, int imageWidth, int imageHeight) Draws the bounding box using the Graphics2D.
float[][]	getProbDist() Returns the probability for each pixel.

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

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

Methods inherited from class java.lang.Object

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

Constructor Detail

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

Method Detail

getProbDist

public float[][] getProbDist()

Returns the probability for each pixel.

Returns:

the probability for each pixel

draw

public void draw(java.awt.Graphics2D g,
                 int imageWidth,
                 int imageHeight)

Draws the bounding box using the Graphics2D.

Specified by:

draw in interface BoundingBox

Overrides:

draw in class Rectangle

Parameters:

g - the Graphics2D object of the image

imageWidth - the width of the image

imageHeight - the height of the image