public class ImageNet extends AbstractImageFolder implements ai.djl.repository.dataset.PreparedDataset
Each image might have different Shapes.
| Modifier and Type | Class and Description |
|---|---|
static class |
ImageNet.Builder
A builder for a
ImageNet. |
AbstractImageFolder.ImageFolderBuilder<T extends AbstractImageFolder.ImageFolderBuilder>flag, items, repository, synset| 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) |
availableSize, get, getSynset, listImagesgetData, getData, randomSplit, sizepublic static ImageNet.Builder builder()
ImageNet.public java.lang.String[] getWordNetIds()
public java.lang.String[] getClassNames()
public java.lang.String[] getClassFull()
public void prepare(ai.djl.util.Progress progress)
prepare in interface ai.djl.repository.dataset.PreparedDatasetprotected java.nio.file.Path getImagePath(java.lang.String key)
getImagePath in class AbstractImageFolder