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
Shape
s.
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
ImageNet.Builder
A 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.Builder
builder()
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.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, matchingTranslatorOptions
-
Methods inherited from class ai.djl.basicdataset.cv.ImageDataset
getImageChannels, getRecordImage
-
Methods inherited from class ai.djl.training.dataset.RandomAccessDataset
getData, getData, getData, getData, newSubDataset, newSubDataset, randomSplit, size, subDataset, subDataset, subDataset, 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:
getImagePath
in classAbstractImageFolder
-
-