org.springframework.cloud.fn.object.detection

Class ObjectDetectionImageAugmenter

java.lang.Object

org.springframework.cloud.fn.object.detection.ObjectDetectionImageAugmenter

All Implemented Interfaces:

BiFunction<byte[],List<ObjectDetection>,byte[]>

public class ObjectDetectionImageAugmenter
extends Object
implements BiFunction<byte[],List<ObjectDetection>,byte[]>

Augment the input image fromMemory detected object bounding boxes and categories. For mask models and withMask set to true it draws the instance segmentation image as well.

Author:

Christian Tzolov

Field Summary

Fields

Modifier and Type	Field and Description
static String	DEFAULT_IMAGE_FORMAT Make checkstyle happy.

Constructor Summary

Constructors

Constructor and Description
ObjectDetectionImageAugmenter()
ObjectDetectionImageAugmenter(boolean withMask)

Method Summary

Modifier and Type	Method and Description
byte[]	apply(byte[] imageBytes, List<ObjectDetection> objectDetections)
String	getImageFormat()
boolean	isAgnosticColors()
void	setAgnosticColors(boolean agnosticColors)
void	setImageFormat(String imageFormat)

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.util.function.BiFunction

andThen

Field Detail

DEFAULT_IMAGE_FORMAT

public static final String DEFAULT_IMAGE_FORMAT

Make checkstyle happy.

See Also:

Constant Field Values

Constructor Detail

ObjectDetectionImageAugmenter

public ObjectDetectionImageAugmenter()

ObjectDetectionImageAugmenter

public ObjectDetectionImageAugmenter(boolean withMask)

Method Detail

isAgnosticColors

public boolean isAgnosticColors()

setAgnosticColors

public void setAgnosticColors(boolean agnosticColors)

getImageFormat

public String getImageFormat()

setImageFormat

public void setImageFormat(String imageFormat)

apply

public byte[] apply(byte[] imageBytes,
                    List<ObjectDetection> objectDetections)

Specified by:

apply in interface BiFunction<byte[],List<ObjectDetection>,byte[]>