Package ai.djl.basicdataset.cv.classification

Class FashionMnist

java.lang.Object
ai.djl.training.dataset.RandomAccessDataset
ai.djl.training.dataset.ArrayDataset
ai.djl.basicdataset.cv.classification.FashionMnist
All Implemented Interfaces:
ai.djl.training.dataset.Dataset
public final class FashionMnist extends ai.djl.training.dataset.ArrayDataset
FashMnist is a dataset from Zalando article images (https://github.com/zalandoresearch/fashion-mnist).

Each sample is a grayscale image (in 3-D NDArray) with shape (28, 28, 1).

It was created to be a drop in replacement for Mnist, but have a less simplistic task.

  • Nested Class Summary

    Nested Classes
    Modifier and Type
    Class
    Description
    static final class 
    FashionMnist.Builder
    A builder for a FashionMnist.

    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 int
    NUM_CLASSES
     

    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 FashionMnist.Builder
    builder()
    Creates a builder to build a Mnist.
    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

  • Method Details

    • builder

      public static FashionMnist.Builder builder()
      Creates a builder to build a Mnist.
      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