org.datavec.image.loader

Class LFWLoader

java.lang.Object

org.datavec.image.loader.BaseImageLoader

org.datavec.image.loader.LFWLoader

All Implemented Interfaces:

java.io.Serializable

public class LFWLoader
extends BaseImageLoader
implements java.io.Serializable

Loads LFW faces data transform. Customize the size of images by passing in preferred dimensions. DataSet 5749 different individuals 1680 people have two or more images in the database 4069 people have just a single image in the database available as 250 by 250 pixel JPEG images most images are in color, although a few are grayscale

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
static int	CHANNELS
static java.lang.String	DATA_URL
java.lang.String	dataFile
protected java.io.File	fullDir
static int	HEIGHT
static PathLabelGenerator	LABEL_PATTERN
static java.lang.String	LABEL_URL
java.lang.String	labelFile
static java.util.Map<java.lang.String,java.lang.String>	lfwData
static java.util.Map<java.lang.String,java.lang.String>	lfwLabel
static java.util.Map<java.lang.String,java.lang.String>	lfwSubsetData
java.lang.String	localDir
java.lang.String	localSubDir
static int	NUM_IMAGES
static int	NUM_LABELS
protected static java.lang.String	REGEX_PATTERN
static int	SUB_NUM_IMAGES
static int	SUB_NUM_LABELS
static java.lang.String	SUBSET_URL
java.lang.String	subsetFile
protected boolean	useSubset
static int	WIDTH

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

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

Constructor Summary

Constructors

Constructor and Description
LFWLoader()
LFWLoader(boolean useSubset)
LFWLoader(int[] imgDim, boolean useSubset)
LFWLoader(int[] imgDim, ImageTransform imgTransform, boolean useSubset)

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.api.ndarray.INDArray	asMatrix(java.io.File f)
org.nd4j.linalg.api.ndarray.INDArray	asMatrix(java.io.InputStream inputStream)
org.nd4j.linalg.api.ndarray.INDArray	asRowVector(java.io.File f)
org.nd4j.linalg.api.ndarray.INDArray	asRowVector(java.io.InputStream inputStream)
void	generateLfwMaps()
RecordReader	getRecordReader(int numExamples)
RecordReader	getRecordReader(int batchSize, int numExamples, boolean train, double splitTrainTest)
RecordReader	getRecordReader(int batchSize, int numExamples, int[] imgDim, boolean train, double splitTrainTest, java.util.Random rng)
RecordReader	getRecordReader(int batchSize, int numExamples, int[] imgDim, int numLabels, PathLabelGenerator labelGenerator, boolean train, double splitTrainTest, java.util.Random rng)
RecordReader	getRecordReader(int batchSize, int numExamples, int[] imgDim, PathLabelGenerator labelGenerator, boolean train, double splitTrainTest, java.util.Random rng)
RecordReader	getRecordReader(int batchSize, int numExamples, int numLabels, java.util.Random rng)
RecordReader	getRecordReader(int batchSize, int numExamples, PathLabelGenerator labelGenerator, boolean train, double splitTrainTest, java.util.Random rng)
boolean	imageFilesExist()
void	load()
void	load(int batchSize, int numExamples, int numLabels, PathLabelGenerator labelGenerator, double splitTrainTest, java.util.Random rng)

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

downloadAndUntar, getAllowedFormats

Methods inherited from class java.lang.Object

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

Field Detail

NUM_IMAGES

public static final int NUM_IMAGES

See Also:

Constant Field Values

NUM_LABELS

public static final int NUM_LABELS

See Also:

Constant Field Values

SUB_NUM_IMAGES

public static final int SUB_NUM_IMAGES

See Also:

Constant Field Values

SUB_NUM_LABELS

public static final int SUB_NUM_LABELS

See Also:

Constant Field Values

HEIGHT

public static final int HEIGHT

See Also:

Constant Field Values

WIDTH

public static final int WIDTH

See Also:

Constant Field Values

CHANNELS

public static final int CHANNELS

See Also:

Constant Field Values

DATA_URL

public static final java.lang.String DATA_URL

See Also:

Constant Field Values

LABEL_URL

public static final java.lang.String LABEL_URL

See Also:

Constant Field Values

SUBSET_URL

public static final java.lang.String SUBSET_URL

See Also:

Constant Field Values

REGEX_PATTERN

protected static final java.lang.String REGEX_PATTERN

See Also:

Constant Field Values

LABEL_PATTERN

public static final PathLabelGenerator LABEL_PATTERN

dataFile

public java.lang.String dataFile

labelFile

public java.lang.String labelFile

subsetFile

public java.lang.String subsetFile

localDir

public java.lang.String localDir

localSubDir

public java.lang.String localSubDir

fullDir

protected java.io.File fullDir

useSubset

protected boolean useSubset

lfwData

public static java.util.Map<java.lang.String,java.lang.String> lfwData

lfwLabel

public static java.util.Map<java.lang.String,java.lang.String> lfwLabel

lfwSubsetData

public static java.util.Map<java.lang.String,java.lang.String> lfwSubsetData

Constructor Detail

LFWLoader

public LFWLoader()

LFWLoader

public LFWLoader(boolean useSubset)

LFWLoader

public LFWLoader(int[] imgDim,
                 boolean useSubset)

LFWLoader

public LFWLoader(int[] imgDim,
                 ImageTransform imgTransform,
                 boolean useSubset)

Method Detail

generateLfwMaps

public void generateLfwMaps()

load

public void load()

load

public void load(int batchSize,
                 int numExamples,
                 int numLabels,
                 PathLabelGenerator labelGenerator,
                 double splitTrainTest,
                 java.util.Random rng)

imageFilesExist

public boolean imageFilesExist()

getRecordReader

public RecordReader getRecordReader(int numExamples)

getRecordReader

public RecordReader getRecordReader(int batchSize,
                                    int numExamples,
                                    int numLabels,
                                    java.util.Random rng)

getRecordReader

public RecordReader getRecordReader(int batchSize,
                                    int numExamples,
                                    boolean train,
                                    double splitTrainTest)

getRecordReader

public RecordReader getRecordReader(int batchSize,
                                    int numExamples,
                                    int[] imgDim,
                                    boolean train,
                                    double splitTrainTest,
                                    java.util.Random rng)

getRecordReader

public RecordReader getRecordReader(int batchSize,
                                    int numExamples,
                                    PathLabelGenerator labelGenerator,
                                    boolean train,
                                    double splitTrainTest,
                                    java.util.Random rng)

getRecordReader

public RecordReader getRecordReader(int batchSize,
                                    int numExamples,
                                    int[] imgDim,
                                    PathLabelGenerator labelGenerator,
                                    boolean train,
                                    double splitTrainTest,
                                    java.util.Random rng)

getRecordReader

public RecordReader getRecordReader(int batchSize,
                                    int numExamples,
                                    int[] imgDim,
                                    int numLabels,
                                    PathLabelGenerator labelGenerator,
                                    boolean train,
                                    double splitTrainTest,
                                    java.util.Random rng)

asRowVector

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

Specified by:

asRowVector in class BaseImageLoader

Throws:

java.io.IOException

asRowVector

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

Specified by:

asRowVector in class BaseImageLoader

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 inputStream)
                                              throws java.io.IOException

Specified by:

asMatrix in class BaseImageLoader

Throws:

java.io.IOException