ai.djl.basicdataset.cv.classification

Class Cifar10

java.lang.Object

ai.djl.training.dataset.RandomAccessDataset

ai.djl.training.dataset.ArrayDataset

ai.djl.basicdataset.cv.classification.Cifar10

All Implemented Interfaces:

ai.djl.training.dataset.Dataset

public final class Cifar10
extends ai.djl.training.dataset.ArrayDataset

CIFAR10 image classification dataset from https://www.cs.toronto.edu/~kriz/cifar.html.

Each sample is an image (in 3-D NDArray) with shape (32, 32, 3).

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	Cifar10.Builder A builder to construct a Cifar10.

Nested classes/interfaces inherited from class ai.djl.training.dataset.RandomAccessDataset

ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<T extends ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<T>>

Nested classes/interfaces inherited from interface ai.djl.training.dataset.Dataset

ai.djl.training.dataset.Dataset.Usage

Field Summary

Fields

Modifier and Type	Field and Description
static int	IMAGE_HEIGHT
static int	IMAGE_WIDTH
static float[]	NORMALIZE_MEAN
static float[]	NORMALIZE_STD

Fields inherited from class ai.djl.training.dataset.ArrayDataset

data, labels

Fields inherited from class ai.djl.training.dataset.RandomAccessDataset

dataBatchifier, device, labelBatchifier, limit, pipeline, prefetchNumber, sampler, targetPipeline

Method Summary

Modifier and Type	Method and Description
static Cifar10.Builder	builder() Creates a builder to build a Cifar10.
void	prepare(ai.djl.util.Progress progress)

Methods inherited from class ai.djl.training.dataset.ArrayDataset

availableSize, get

Methods inherited from class ai.djl.training.dataset.RandomAccessDataset

getData, getData, getData, getData, randomSplit, size, subDataset, toArray

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.training.dataset.Dataset

prepare

Field Detail

IMAGE_WIDTH

public static final int IMAGE_WIDTH

See Also:

Constant Field Values

IMAGE_HEIGHT

public static final int IMAGE_HEIGHT

See Also:

Constant Field Values

NORMALIZE_MEAN

public static final float[] NORMALIZE_MEAN

NORMALIZE_STD

public static final float[] NORMALIZE_STD

Method Detail

builder

public static Cifar10.Builder builder()

Creates a builder to build a Cifar10.

Returns:

a new builder

prepare

public void prepare(ai.djl.util.Progress progress)
             throws java.io.IOException

Specified by:

prepare in interface ai.djl.training.dataset.Dataset

Overrides:

prepare in class ai.djl.training.dataset.ArrayDataset

Throws:

java.io.IOException