org.datavec.image.transform

Class PipelineImageTransform

java.lang.Object

org.datavec.image.transform.BaseImageTransform<org.bytedeco.javacpp.opencv_core.Mat>

org.datavec.image.transform.PipelineImageTransform

All Implemented Interfaces:

ImageTransform

public class PipelineImageTransform
extends BaseImageTransform<org.bytedeco.javacpp.opencv_core.Mat>

Allows creation of image transform pipelines, either sequentially or randomly. You have the option of passing in multiple transforms as parameters. If you want to create a more complex pipeline, you can pass in a pipeline that looks like List<Pair<ImageTransform, Double>>. The Double value is the probability that particular element in the pipeline will be executed. This is helpful for creating complex pipelines. The pipeline can also be randomly shuffled with each transform, further increasing the available dataset.

Author:

saudet, crockpotveggies

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	PipelineImageTransform.Builder Optional builder helper for PipelineImageTransform

Field Summary

Fields

Modifier and Type	Field and Description
protected List<org.nd4j.linalg.primitives.Pair<ImageTransform,Double>>	imageTransforms
protected org.nd4j.linalg.api.rng.Random	rng
protected boolean	shuffle

Fields inherited from class org.datavec.image.transform.BaseImageTransform

converter, random

Constructor Summary

Constructors

Constructor and Description
PipelineImageTransform(ImageTransform... transforms)
PipelineImageTransform(List<org.nd4j.linalg.primitives.Pair<ImageTransform,Double>> transforms)
PipelineImageTransform(List<org.nd4j.linalg.primitives.Pair<ImageTransform,Double>> transforms, boolean shuffle)
PipelineImageTransform(long seed, boolean shuffle, ImageTransform... transforms)
PipelineImageTransform(long seed, List<org.nd4j.linalg.primitives.Pair<ImageTransform,Double>> transforms)
PipelineImageTransform(long seed, List<org.nd4j.linalg.primitives.Pair<ImageTransform,Double>> transforms, boolean shuffle)
PipelineImageTransform(Random random, long seed, List<org.nd4j.linalg.primitives.Pair<ImageTransform,Double>> transforms, boolean shuffle)

Method Summary

Modifier and Type	Method and Description
ImageWritable	transform(ImageWritable image, Random random) Takes an image and executes a pipeline of combined transforms.

Methods inherited from class org.datavec.image.transform.BaseImageTransform

transform

Methods inherited from class java.lang.Object

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

Field Detail

imageTransforms

protected List<org.nd4j.linalg.primitives.Pair<ImageTransform,Double>> imageTransforms

shuffle

protected boolean shuffle

rng

protected org.nd4j.linalg.api.rng.Random rng

Constructor Detail

PipelineImageTransform

public PipelineImageTransform(ImageTransform... transforms)

PipelineImageTransform

public PipelineImageTransform(long seed,
                              boolean shuffle,
                              ImageTransform... transforms)

PipelineImageTransform

public PipelineImageTransform(List<org.nd4j.linalg.primitives.Pair<ImageTransform,Double>> transforms)

PipelineImageTransform

public PipelineImageTransform(List<org.nd4j.linalg.primitives.Pair<ImageTransform,Double>> transforms,
                              boolean shuffle)

PipelineImageTransform

public PipelineImageTransform(long seed,
                              List<org.nd4j.linalg.primitives.Pair<ImageTransform,Double>> transforms)

PipelineImageTransform

public PipelineImageTransform(long seed,
                              List<org.nd4j.linalg.primitives.Pair<ImageTransform,Double>> transforms,
                              boolean shuffle)

PipelineImageTransform

public PipelineImageTransform(Random random,
                              long seed,
                              List<org.nd4j.linalg.primitives.Pair<ImageTransform,Double>> transforms,
                              boolean shuffle)

Method Detail

transform

public ImageWritable transform(ImageWritable image,
                               Random random)

Takes an image and executes a pipeline of combined transforms.

Parameters:

image - to transform, null == end of stream

random - object to use (or null for deterministic)

Returns:

transformed image