public final class ImageFolder extends AbstractImageFolder
The image folder should be structured as follows:
root/shoes/Aerobic Shoes1.png
root/shoes/Aerobic Shose2.png
...
root/boots/Black Boots.png
root/boots/White Boots.png
...
root/pumps/Red Pumps
root/pumps/Pink Pumps
...
here shoes, boots, pumps are your labels
| Modifier and Type | Class and Description |
|---|---|
static class |
ImageFolder.Builder
A builder for the
ImageFolder. |
AbstractImageFolder.ImageFolderBuilder<T extends AbstractImageFolder.ImageFolderBuilder>flag, items, repository, synset| Modifier and Type | Method and Description |
|---|---|
static ImageFolder.Builder |
builder()
Creates a new builder to build a
ImageFolder. |
protected java.nio.file.Path |
getImagePath(java.lang.String key) |
void |
prepare(ai.djl.util.Progress progress) |
availableSize, get, getSynset, listImagesgetData, getData, randomSplit, sizepublic static ImageFolder.Builder builder()
ImageFolder.protected java.nio.file.Path getImagePath(java.lang.String key)
getImagePath in class AbstractImageFolderpublic void prepare(ai.djl.util.Progress progress)