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<T>>
ImageDataset.BaseBuilder<T extends ImageDataset.BaseBuilder<T>>
items, prepared, resource, synset
flag
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, getClasses, getClassNumber, getImage, getImageHeight, getImageWidth, getSynset, listImages
get, makeTranslator
getImageChannels, getRecordImage
getData, getData, getData, getData, randomSplit, size, subDataset, toArray
public static ImageFolder.Builder builder()
ImageFolder
.protected java.nio.file.Path getImagePath(java.lang.String key)
getImagePath
in class AbstractImageFolder
public void prepare(ai.djl.util.Progress progress) throws java.io.IOException
java.io.IOException