ai.djl.modality.cv.util

Class BufferedImageUtils

java.lang.Object

ai.djl.modality.cv.util.BufferedImageUtils

public final class BufferedImageUtils
extends java.lang.Object

BufferedImageUtils is an image processing utility to load, reshape and convert images using BufferedImage.

Method Summary

Modifier and Type	Method and Description
static java.awt.image.BufferedImage	fromFile(java.nio.file.Path path) Loads the image from the specified file.
static java.awt.image.BufferedImage	fromUrl(java.lang.String url) Loads the image from the specified URL.
static java.awt.image.BufferedImage	fromUrl(java.net.URL url) Loads the image from the specified URL.
static java.awt.Color	randomColor() Returns a random color.
static NDArray	readFileToArray(NDManager manager, java.nio.file.Path path) Reads an image file and converts to NDArray.
static NDArray	readFileToArray(NDManager manager, java.nio.file.Path path, NDImageUtils.Flag flag) Reads an image file and converts to NDArray.
static NDArray	toNDArray(NDManager manager, java.awt.image.BufferedImage image) Converts a BufferedImage to an RGB NDArray.
static NDArray	toNDArray(NDManager manager, java.awt.image.BufferedImage image, NDImageUtils.Flag flag) Converts a BufferedImage to an NDArray with designated color mode.

Methods inherited from class java.lang.Object

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

Method Detail

fromFile

public static java.awt.image.BufferedImage fromFile(java.nio.file.Path path)
                                             throws java.io.IOException

Loads the image from the specified file.

Parameters:

path - the path of the file to be loaded

Returns:

a BufferedImage

Throws:

java.io.IOException - if file is not found

fromUrl

public static java.awt.image.BufferedImage fromUrl(java.net.URL url)
                                            throws java.io.IOException

Loads the image from the specified URL.

Parameters:

url - the URL of the file to be loaded

Returns:

a BufferedImage

Throws:

java.io.IOException - if url is not found

fromUrl

public static java.awt.image.BufferedImage fromUrl(java.lang.String url)
                                            throws java.io.IOException

Loads the image from the specified URL.

Parameters:

url - the URL of the file to be loaded

Returns:

a BufferedImage

Throws:

java.io.IOException - if url is not found

randomColor

public static java.awt.Color randomColor()

Returns a random color.

Returns:

a random color

toNDArray

public static NDArray toNDArray(NDManager manager,
                                java.awt.image.BufferedImage image)

Converts a BufferedImage to an RGB NDArray.

Parameters:

manager - a NDManager to create the new NDArray with

image - the buffered image to be converted

Returns:

the image NDArray

toNDArray

public static NDArray toNDArray(NDManager manager,
                                java.awt.image.BufferedImage image,
                                NDImageUtils.Flag flag)

Converts a BufferedImage to an NDArray with designated color mode.

Parameters:

manager - a NDManager to create the new NDArray with

image - the buffered image to be converted

flag - the color mode

Returns:

the image NDArray

readFileToArray

public static NDArray readFileToArray(NDManager manager,
                                      java.nio.file.Path path)
                               throws java.io.IOException

Reads an image file and converts to NDArray.

Returns an image with Color mode.

Parameters:

manager - a NDManager to create the new NDArray with

path - the path to the image file

Returns:

the image NDArray

Throws:

java.io.IOException - if file is not found

readFileToArray

public static NDArray readFileToArray(NDManager manager,
                                      java.nio.file.Path path,
                                      NDImageUtils.Flag flag)
                               throws java.io.IOException

Reads an image file and converts to NDArray.

Parameters:

manager - a NDManager to create the new NDArray with

path - the path to the image file

flag - the color option to load with

Returns:

the image NDArray

Throws:

java.io.IOException - if file is not found