org.datavec.image.loader

Class NativeImageLoader

java.lang.Object

org.datavec.image.loader.BaseImageLoader

org.datavec.image.loader.NativeImageLoader

All Implemented Interfaces:

java.io.Serializable

Direct Known Subclasses:

CifarLoader

public class NativeImageLoader
extends BaseImageLoader

Uses JavaCV to load images. Allowed formats: bmp, gif, jpg, jpeg, jp2, pbm, pgm, ppm, pnm, png, tif, tiff, exr, webp

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
static java.lang.String[]	ALLOWED_FORMATS

Fields inherited from class org.datavec.image.loader.BaseImageLoader

BASE_DIR, centerCropIfNeeded, channels, height, imageTransform, log, rng, width

Constructor Summary

Constructors

Constructor and Description
NativeImageLoader() Loads images with no scaling or conversion.
NativeImageLoader(int height, int width) Instantiate an image with the given height and width
NativeImageLoader(int height, int width, int channels) Instantiate an image with the given height and width
NativeImageLoader(int height, int width, int channels, boolean centerCropIfNeeded) Instantiate an image with the given height and width
NativeImageLoader(int height, int width, int channels, ImageTransform imageTransform) Instantiate an image with the given height and width

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.api.ndarray.INDArray	asMatrix(java.awt.image.BufferedImage image)
org.nd4j.linalg.api.ndarray.INDArray	asMatrix(java.io.File f)
org.nd4j.linalg.api.ndarray.INDArray	asMatrix(java.io.InputStream is)
org.nd4j.linalg.api.ndarray.INDArray	asMatrix(org.bytedeco.javacpp.opencv_core.Mat image)
org.nd4j.linalg.api.ndarray.INDArray	asRowVector(java.awt.image.BufferedImage image)
org.nd4j.linalg.api.ndarray.INDArray	asRowVector(java.io.File f) Convert a file to a row vector
org.nd4j.linalg.api.ndarray.INDArray	asRowVector(java.io.InputStream is)
org.nd4j.linalg.api.ndarray.INDArray	asRowVector(org.bytedeco.javacpp.opencv_core.Mat image)
protected org.bytedeco.javacpp.opencv_core.Mat	centerCropIfNeeded(org.bytedeco.javacpp.opencv_core.Mat img)
java.lang.String[]	getAllowedFormats()
protected org.bytedeco.javacpp.opencv_core.Mat	scalingIfNeed(org.bytedeco.javacpp.opencv_core.Mat image)
protected org.bytedeco.javacpp.opencv_core.Mat	scalingIfNeed(org.bytedeco.javacpp.opencv_core.Mat image, int dstHeight, int dstWidth)

Methods inherited from class org.datavec.image.loader.BaseImageLoader

downloadAndUntar

Methods inherited from class java.lang.Object

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

Field Detail

ALLOWED_FORMATS

public static final java.lang.String[] ALLOWED_FORMATS

Constructor Detail

NativeImageLoader

public NativeImageLoader()

Loads images with no scaling or conversion.

NativeImageLoader

public NativeImageLoader(int height,
                         int width)

Instantiate an image with the given height and width

Parameters:

height - the height to load*

width - the width to load

NativeImageLoader

public NativeImageLoader(int height,
                         int width,
                         int channels)

Instantiate an image with the given height and width

Parameters:

height - the height to load

width - the width to load

channels - the number of channels for the image*

NativeImageLoader

public NativeImageLoader(int height,
                         int width,
                         int channels,
                         boolean centerCropIfNeeded)

Instantiate an image with the given height and width

Parameters:

height - the height to load

width - the width to load

channels - the number of channels for the image*

centerCropIfNeeded - to crop before rescaling and converting

NativeImageLoader

public NativeImageLoader(int height,
                         int width,
                         int channels,
                         ImageTransform imageTransform)

Instantiate an image with the given height and width

Parameters:

height - the height to load

width - the width to load

channels - the number of channels for the image*

imageTransform - to use before rescaling and converting

Method Detail

getAllowedFormats

public java.lang.String[] getAllowedFormats()

Overrides:

getAllowedFormats in class BaseImageLoader

asRowVector

public org.nd4j.linalg.api.ndarray.INDArray asRowVector(java.io.File f)
                                                 throws java.io.IOException

Convert a file to a row vector

Specified by:

asRowVector in class BaseImageLoader

Parameters:

f - the image to convert

Returns:

the flattened image

Throws:

java.io.IOException

asRowVector

public org.nd4j.linalg.api.ndarray.INDArray asRowVector(java.io.InputStream is)
                                                 throws java.io.IOException

Specified by:

asRowVector in class BaseImageLoader

Throws:

java.io.IOException

asRowVector

public org.nd4j.linalg.api.ndarray.INDArray asRowVector(java.awt.image.BufferedImage image)
                                                 throws java.io.IOException

Throws:

java.io.IOException

asRowVector

public org.nd4j.linalg.api.ndarray.INDArray asRowVector(org.bytedeco.javacpp.opencv_core.Mat image)
                                                 throws java.io.IOException

Throws:

java.io.IOException

asMatrix

public org.nd4j.linalg.api.ndarray.INDArray asMatrix(java.io.File f)
                                              throws java.io.IOException

Specified by:

asMatrix in class BaseImageLoader

Throws:

java.io.IOException

asMatrix

public org.nd4j.linalg.api.ndarray.INDArray asMatrix(java.io.InputStream is)
                                              throws java.io.IOException

Specified by:

asMatrix in class BaseImageLoader

Throws:

java.io.IOException

asMatrix

public org.nd4j.linalg.api.ndarray.INDArray asMatrix(java.awt.image.BufferedImage image)
                                              throws java.io.IOException

Throws:

java.io.IOException

asMatrix

public org.nd4j.linalg.api.ndarray.INDArray asMatrix(org.bytedeco.javacpp.opencv_core.Mat image)
                                              throws java.io.IOException

Throws:

java.io.IOException

centerCropIfNeeded

protected org.bytedeco.javacpp.opencv_core.Mat centerCropIfNeeded(org.bytedeco.javacpp.opencv_core.Mat img)

scalingIfNeed

protected org.bytedeco.javacpp.opencv_core.Mat scalingIfNeed(org.bytedeco.javacpp.opencv_core.Mat image)

scalingIfNeed

protected org.bytedeco.javacpp.opencv_core.Mat scalingIfNeed(org.bytedeco.javacpp.opencv_core.Mat image,
                                                             int dstHeight,
                                                             int dstWidth)