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,java.io.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.
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description float[][]getProbDist()Returns the probability for each pixel.
-
-
-
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 rectangley- the top coordinate of the bounding rectanglewidth- the width of the bounding rectangleheight- the height of the bounding rectangledist- the probability distribution for each pixel in the rectangle
-
-