ai.djl.basicdataset

Class ImageNet

java.lang.Object

ai.djl.training.dataset.RandomAccessDataset

ai.djl.basicdataset.AbstractImageFolder

ai.djl.basicdataset.ImageNet

All Implemented Interfaces:

ai.djl.repository.dataset.PreparedDataset, ai.djl.training.dataset.Dataset, java.util.RandomAccess

public class ImageNet
extends AbstractImageFolder
implements ai.djl.repository.dataset.PreparedDataset

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

AbstractImageFolder.ImageFolderBuilder<T extends AbstractImageFolder.ImageFolderBuilder>

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>

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

flag, items, repository, synset

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

batchifier, device, executor, 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.AbstractImageFolder

availableSize, get, getSynset, listImages

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

getData, getData, randomSplit, size

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.repository.dataset.PreparedDataset

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)

Specified by:

prepare in interface ai.djl.repository.dataset.PreparedDataset

getImagePath

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

Specified by:

getImagePath in class AbstractImageFolder