org.deeplearning4j.datasets.fetchers

Class Cifar10Fetcher

java.lang.Object

org.deeplearning4j.datasets.fetchers.CacheableExtractableDataSetFetcher

org.deeplearning4j.datasets.fetchers.Cifar10Fetcher

public class Cifar10Fetcher
extends CacheableExtractableDataSetFetcher

CifarDataSetIterator is an iterator for CIFAR-10 dataset - 10 classes, with 32x32 images with 3 channels (RGB) This fetcher uses a cached version of the CIFAR dataset which is converted to PNG images, see: https://pjreddie.com/projects/cifar-10-dataset-mirror/.

Author:

Justin Long (crockpotveggies)

Field Summary

Fields

Modifier and Type	Field and Description
static int	INPUT_CHANNELS
static int	INPUT_HEIGHT
static int	INPUT_WIDTH
static String	LABELS_FILENAME
static String	LOCAL_CACHE_NAME
static int	NUM_LABELS

Constructor Summary

Constructors

Constructor and Description
Cifar10Fetcher()

Method Summary

Modifier and Type	Method and Description
long	expectedChecksum(DataSetType set)
org.datavec.api.records.reader.RecordReader	getRecordReader(long rngSeed, int[] imgDim, DataSetType set, ImageTransform imageTransform)
String	localCacheName()
String	remoteDataUrl(DataSetType set)

Methods inherited from class org.deeplearning4j.datasets.fetchers.CacheableExtractableDataSetFetcher

dataSetName, deleteIfEmpty, downloadAndExtract, downloadAndExtract, expectedChecksum, getLocalCacheDir, isCached, remoteDataUrl

Methods inherited from class java.lang.Object

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

Field Detail

LABELS_FILENAME

public static final String LABELS_FILENAME

See Also:

Constant Field Values

LOCAL_CACHE_NAME

public static final String LOCAL_CACHE_NAME

See Also:

Constant Field Values

INPUT_WIDTH

public static int INPUT_WIDTH

INPUT_HEIGHT

public static int INPUT_HEIGHT

INPUT_CHANNELS

public static int INPUT_CHANNELS

NUM_LABELS

public static int NUM_LABELS

Constructor Detail

Cifar10Fetcher

public Cifar10Fetcher()

Method Detail

remoteDataUrl

public String remoteDataUrl(DataSetType set)

localCacheName

public String localCacheName()

expectedChecksum

public long expectedChecksum(DataSetType set)

getRecordReader

public org.datavec.api.records.reader.RecordReader getRecordReader(long rngSeed,
                                                                   int[] imgDim,
                                                                   DataSetType set,
                                                                   ImageTransform imageTransform)