org.datavec.image.loader

Class CifarLoader

java.lang.Object

org.datavec.image.loader.BaseImageLoader

org.datavec.image.loader.NativeImageLoader

org.datavec.image.loader.CifarLoader

All Implemented Interfaces:

Serializable

public class CifarLoader
extends NativeImageLoader
implements Serializable

Reference: Learning Multiple Layers of Features from Tiny Images, Alex Krizhevsky, 2009. Created by nyghtowl on 12/17/15.

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
static int	BYTEFILELEN
static int	CHANNELS
static Map<String,String>	cifarTrainData
static String	dataBinFile
static String	dataBinUrl
String	dataFile
String	dataUrl
protected static File	fullDir
static int	HEIGHT
protected static String	labelFileName
static File	LABELPATH
protected static List<String>	labels
static String	localDir
static int	NUM_LABELS
static int	NUM_TEST_IMAGES
static int	NUM_TRAIN_IMAGES
protected static String	testFileName
static File	TESTPATH
protected boolean	train
protected static String[]	trainFileNames
static File	TRAINPATH
static int	WIDTH

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

ALLOWED_FORMATS

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

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

Constructor Summary

Constructors

Constructor and Description
CifarLoader()
CifarLoader(boolean train)
CifarLoader(boolean train, String localDir)
CifarLoader(int height, int width, int channels)
CifarLoader(int height, int width, int channels, boolean train)
CifarLoader(int height, int width, int channels, ImageTransform imgTransform, int normalizeValue, boolean train)

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) Convert a file to a row vector
org.nd4j.linalg.api.ndarray.INDArray	asRowVector(InputStream inputStream)
static void	generateMaps()
InputStream	getInputStream()
List<String>	getLabels()
static boolean	imageFilesExist()
static void	load()

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

asMatrix, asRowVector, centerCropIfNeeded, getAllowedFormats, normalizeIfNeeded, scalingIfNeed, scalingIfNeed

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

NUM_TRAIN_IMAGES

public static final int NUM_TRAIN_IMAGES

See Also:

Constant Field Values

NUM_TEST_IMAGES

public static final int NUM_TEST_IMAGES

See Also:

Constant Field Values

NUM_LABELS

public static final int 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

BYTEFILELEN

public static final int BYTEFILELEN

See Also:

Constant Field Values

dataUrl

public String dataUrl

dataFile

public String dataFile

dataBinUrl

public static String dataBinUrl

dataBinFile

public static String dataBinFile

labelFileName

protected static String labelFileName

labels

protected static List<String> labels

trainFileNames

protected static String[] trainFileNames

testFileName

protected static String testFileName

localDir

public static String localDir

fullDir

protected static File fullDir

train

protected boolean train

cifarTrainData

public static Map<String,String> cifarTrainData

TRAINPATH

public static final File TRAINPATH

TESTPATH

public static final File TESTPATH

LABELPATH

public static final File LABELPATH

Constructor Detail

CifarLoader

public CifarLoader(boolean train)

CifarLoader

public CifarLoader(int height,
                   int width,
                   int channels,
                   boolean train)

CifarLoader

public CifarLoader(int height,
                   int width,
                   int channels,
                   ImageTransform imgTransform,
                   int normalizeValue,
                   boolean train)

CifarLoader

public CifarLoader(boolean train,
                   String localDir)

CifarLoader

public CifarLoader()

CifarLoader

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

Method Detail

asRowVector

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

Description copied from class: NativeImageLoader

Convert a file to a row vector

Overrides:

asRowVector in class NativeImageLoader

Parameters:

f - the image to convert

Returns:

the flattened image

Throws:

IOException

asRowVector

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

Overrides:

asRowVector in class NativeImageLoader

Throws:

IOException

asMatrix

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

Overrides:

asMatrix in class NativeImageLoader

Throws:

IOException

asMatrix

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

Overrides:

asMatrix in class NativeImageLoader

Throws:

IOException

generateMaps

public static void generateMaps()

load

public static void load()

imageFilesExist

public static boolean imageFilesExist()

getInputStream

public InputStream getInputStream()

getLabels

public List<String> getLabels()