ai.djl.basicdataset.cv.classification

Class ImageNet

java.lang.Object

ai.djl.training.dataset.RandomAccessDataset

ai.djl.basicdataset.cv.ImageDataset

ai.djl.basicdataset.cv.classification.ImageClassificationDataset

ai.djl.basicdataset.cv.classification.AbstractImageFolder

ai.djl.basicdataset.cv.classification.ImageNet

All Implemented Interfaces:

ai.djl.training.dataset.Dataset

public class ImageNet
extends AbstractImageFolder

ImageNet is an image classification dataset from http://image-net.org 2012 Classification dataset.

Each image might have different Shapes.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	ImageNet.Builder A builder for a ImageNet.

Nested classes/interfaces inherited from class ai.djl.basicdataset.cv.classification.AbstractImageFolder

AbstractImageFolder.ImageFolderBuilder<T extends AbstractImageFolder.ImageFolderBuilder<T>>

Nested classes/interfaces inherited from class ai.djl.basicdataset.cv.ImageDataset

ImageDataset.BaseBuilder<T extends ImageDataset.BaseBuilder<T>>

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

ai.djl.training.dataset.Dataset.Usage

Field Summary

Fields inherited from class ai.djl.basicdataset.cv.classification.AbstractImageFolder

items, mrl, prepared, synset

Fields inherited from class ai.djl.basicdataset.cv.ImageDataset

flag

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 ImageNet.Builder	builder() Creates a new builder to build a ImageNet.
java.lang.String[]	getClassFull() Returns all full class names of this ImageNet dataset.
java.lang.String[]	getClassNames() Returns all class names of this ImageNet dataset.
protected java.nio.file.Path	getImagePath(java.lang.String key)
java.lang.String[]	getWordNetIds() Returns all WordNet ids of this ImageNet dataset.
void	prepare(ai.djl.util.Progress progress)

Methods inherited from class ai.djl.basicdataset.cv.classification.AbstractImageFolder

availableSize, getClasses, getClassNumber, getImage, getImageHeight, getImageWidth, getSynset, listImages

Methods inherited from class ai.djl.basicdataset.cv.classification.ImageClassificationDataset

get, makeTranslator

Methods inherited from class ai.djl.basicdataset.cv.ImageDataset

getImageChannels, getRecordImage

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

Method Detail

builder

public static ImageNet.Builder builder()

Creates a new builder to build a ImageNet.

Returns:

a new builder

getWordNetIds

public java.lang.String[] getWordNetIds()

Returns all WordNet ids of this ImageNet dataset.

Returns:

all WordNet ids of this ImageNet dataset

getClassNames

public java.lang.String[] getClassNames()

Returns all class names of this ImageNet dataset.

Returns:

all class names of this ImageNet dataset

getClassFull

public java.lang.String[] getClassFull()

Returns all full class names of this ImageNet dataset.

Returns:

all full class names of this ImageNet dataset

prepare

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

Throws:

java.io.IOException

getImagePath

protected java.nio.file.Path getImagePath(java.lang.String key)

Specified by:

getImagePath in class AbstractImageFolder