org.datavec.image.loader

Class NativeImageLoader

java.lang.Object

org.datavec.image.loader.BaseImageLoader

org.datavec.image.loader.NativeImageLoader

All Implemented Interfaces:

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

Author:

saudet

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
static 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(BufferedImage image)
org.nd4j.linalg.api.ndarray.INDArray	asMatrix(File f)
org.nd4j.linalg.api.ndarray.INDArray	asMatrix(InputStream is)
org.nd4j.linalg.api.ndarray.INDArray	asMatrix(org.bytedeco.javacpp.opencv_core.Mat image)
org.nd4j.linalg.api.ndarray.INDArray	asRowVector(BufferedImage image)
org.nd4j.linalg.api.ndarray.INDArray	asRowVector(File f) Convert a file to a row vector
org.nd4j.linalg.api.ndarray.INDArray	asRowVector(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)
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 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 String[] getAllowedFormats()

Overrides:

getAllowedFormats in class BaseImageLoader

asRowVector

public org.nd4j.linalg.api.ndarray.INDArray asRowVector(File f)
                                                 throws 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:

IOException

asRowVector

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

Specified by:

asRowVector in class BaseImageLoader

Throws:

IOException

asRowVector

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

Throws:

IOException

asRowVector

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

Throws:

IOException

asMatrix

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

Specified by:

asMatrix in class BaseImageLoader

Throws:

IOException

asMatrix

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

Specified by:

asMatrix in class BaseImageLoader

Throws:

IOException

asMatrix

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

Throws:

IOException

asMatrix

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

Throws:

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)