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 Description static classImageNet.BuilderA builder for aImageNet.-
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>>
-
-
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
-
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description static ImageNet.Builderbuilder()Creates a new builder to build aImageNet.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.PathgetImagePath(java.lang.String key)java.lang.String[]getWordNetIds()Returns all WordNet ids of this ImageNet dataset.voidprepare(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
-
-
-
-
Method Detail
-
builder
public static ImageNet.Builder builder()
Creates a new builder to build aImageNet.- 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:
getImagePathin classAbstractImageFolder
-
-