org.datavec.image.loader

Class LFWLoader

java.lang.Object

org.datavec.image.loader.BaseImageLoader

org.datavec.image.loader.LFWLoader

All Implemented Interfaces:

Serializable

public class LFWLoader
extends BaseImageLoader
implements 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 String	DATA_URL
String	dataFile
protected File	fullDir
static int	HEIGHT
static org.datavec.api.io.labels.PathLabelGenerator	LABEL_PATTERN
static String	LABEL_URL
String	labelFile
static Map<String,String>	lfwData
static Map<String,String>	lfwLabel
static Map<String,String>	lfwSubsetData
String	localDir
String	localSubDir
static int	NUM_IMAGES
static int	NUM_LABELS
protected static String	REGEX_PATTERN
static int	SUB_NUM_IMAGES
static int	SUB_NUM_LABELS
static String	SUBSET_URL
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(File f)
org.nd4j.linalg.api.ndarray.INDArray	asMatrix(InputStream inputStream)
org.nd4j.linalg.api.ndarray.INDArray	asRowVector(File f)
org.nd4j.linalg.api.ndarray.INDArray	asRowVector(InputStream inputStream)
void	generateLfwMaps()
org.datavec.api.records.reader.RecordReader	getRecordReader(int numExamples)
org.datavec.api.records.reader.RecordReader	getRecordReader(int batchSize, int numExamples, boolean train, double splitTrainTest)
org.datavec.api.records.reader.RecordReader	getRecordReader(int batchSize, int numExamples, int[] imgDim, boolean train, double splitTrainTest, Random rng)
org.datavec.api.records.reader.RecordReader	getRecordReader(int batchSize, int numExamples, int[] imgDim, int numLabels, org.datavec.api.io.labels.PathLabelGenerator labelGenerator, boolean train, double splitTrainTest, Random rng)
org.datavec.api.records.reader.RecordReader	getRecordReader(int batchSize, int numExamples, int[] imgDim, org.datavec.api.io.labels.PathLabelGenerator labelGenerator, boolean train, double splitTrainTest, Random rng)
org.datavec.api.records.reader.RecordReader	getRecordReader(int batchSize, int numExamples, int numLabels, Random rng)
org.datavec.api.records.reader.RecordReader	getRecordReader(int batchSize, int numExamples, org.datavec.api.io.labels.PathLabelGenerator labelGenerator, boolean train, double splitTrainTest, Random rng)
boolean	imageFilesExist()
void	load()
void	load(int batchSize, int numExamples, int numLabels, org.datavec.api.io.labels.PathLabelGenerator labelGenerator, double splitTrainTest, 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 String DATA_URL

See Also:

Constant Field Values

LABEL_URL

public static final String LABEL_URL

See Also:

Constant Field Values

SUBSET_URL

public static final String SUBSET_URL

See Also:

Constant Field Values

REGEX_PATTERN

protected static final String REGEX_PATTERN

See Also:

Constant Field Values

LABEL_PATTERN

public static final org.datavec.api.io.labels.PathLabelGenerator LABEL_PATTERN

dataFile

public String dataFile

labelFile

public String labelFile

subsetFile

public String subsetFile

localDir

public String localDir

localSubDir

public String localSubDir

fullDir

protected File fullDir

useSubset

protected boolean useSubset

lfwData

public static Map<String,String> lfwData

lfwLabel

public static Map<String,String> lfwLabel

lfwSubsetData

public static Map<String,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,
                 org.datavec.api.io.labels.PathLabelGenerator labelGenerator,
                 double splitTrainTest,
                 Random rng)

imageFilesExist

public boolean imageFilesExist()

getRecordReader

public org.datavec.api.records.reader.RecordReader getRecordReader(int numExamples)

getRecordReader

public org.datavec.api.records.reader.RecordReader getRecordReader(int batchSize,
                                                                   int numExamples,
                                                                   int numLabels,
                                                                   Random rng)

getRecordReader

public org.datavec.api.records.reader.RecordReader getRecordReader(int batchSize,
                                                                   int numExamples,
                                                                   boolean train,
                                                                   double splitTrainTest)

getRecordReader

public org.datavec.api.records.reader.RecordReader getRecordReader(int batchSize,
                                                                   int numExamples,
                                                                   int[] imgDim,
                                                                   boolean train,
                                                                   double splitTrainTest,
                                                                   Random rng)

getRecordReader

public org.datavec.api.records.reader.RecordReader getRecordReader(int batchSize,
                                                                   int numExamples,
                                                                   org.datavec.api.io.labels.PathLabelGenerator labelGenerator,
                                                                   boolean train,
                                                                   double splitTrainTest,
                                                                   Random rng)

getRecordReader

public org.datavec.api.records.reader.RecordReader getRecordReader(int batchSize,
                                                                   int numExamples,
                                                                   int[] imgDim,
                                                                   org.datavec.api.io.labels.PathLabelGenerator labelGenerator,
                                                                   boolean train,
                                                                   double splitTrainTest,
                                                                   Random rng)

getRecordReader

public org.datavec.api.records.reader.RecordReader getRecordReader(int batchSize,
                                                                   int numExamples,
                                                                   int[] imgDim,
                                                                   int numLabels,
                                                                   org.datavec.api.io.labels.PathLabelGenerator labelGenerator,
                                                                   boolean train,
                                                                   double splitTrainTest,
                                                                   Random rng)

asRowVector

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

Specified by:

asRowVector in class BaseImageLoader

Throws:

IOException

asRowVector

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

Specified by:

asRowVector in class BaseImageLoader

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

Specified by:

asMatrix in class BaseImageLoader

Throws:

IOException