Package 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.

It consists of 60,000 32x32 color images with 10 classes. It can train in a few hours with a GPU.

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

  • Nested Class Summary

    Nested Classes
    Modifier and Type
    Class
    Description
    static final 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
    Description
    static final int
    IMAGE_HEIGHT
     
    static final int
    IMAGE_WIDTH
     
    static final float[]
    NORMALIZE_MEAN
     
    static final 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
    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, getByIndices, getByRange, getData, newSubDataset, newSubDataset

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

    getData, getData, getData, randomSplit, size, subDataset, subDataset, subDataset, 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

    matchingTranslatorOptions, prepare

  • Field Details

    • IMAGE_WIDTH

      public static final int IMAGE_WIDTH
      See Also:

    • IMAGE_HEIGHT

      public static final int IMAGE_HEIGHT
      See Also:

    • NORMALIZE_MEAN

      public static final float[] NORMALIZE_MEAN

    • NORMALIZE_STD

      public static final float[] NORMALIZE_STD

  • Method Details

    • 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 IOException
      Specified by:
      prepare in interface ai.djl.training.dataset.Dataset
      Overrides:
      prepare in class ai.djl.training.dataset.ArrayDataset
      Throws:
      IOException