ai.djl.basicdataset

Class ImageFolder

java.lang.Object

ai.djl.training.dataset.RandomAccessDataset

ai.djl.basicdataset.AbstractImageFolder

ai.djl.basicdataset.ImageFolder

All Implemented Interfaces:

ai.djl.repository.dataset.PreparedDataset, ai.djl.training.dataset.Dataset, java.util.RandomAccess

public final class ImageFolder
extends AbstractImageFolder

A dataset for loading image files stored in a folder structure.

  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
  

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	ImageFolder.Builder A builder for the ImageFolder.

Nested classes/interfaces inherited from class ai.djl.basicdataset.AbstractImageFolder

AbstractImageFolder.ImageFolderBuilder<T extends AbstractImageFolder.ImageFolderBuilder>

Nested classes/interfaces inherited from class ai.djl.training.dataset.RandomAccessDataset

ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<T extends ai.djl.training.dataset.RandomAccessDataset.BaseBuilder>

Nested classes/interfaces inherited from interface ai.djl.training.dataset.Dataset

ai.djl.training.dataset.Dataset.Usage

Field Summary

Fields inherited from class ai.djl.basicdataset.AbstractImageFolder

flag, items, repository, synset

Fields inherited from class ai.djl.training.dataset.RandomAccessDataset

batchifier, device, executor, limit, pipeline, prefetchNumber, sampler, targetPipeline

Method Summary

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)

Methods inherited from class ai.djl.basicdataset.AbstractImageFolder

availableSize, get, getSynset, listImages

Methods inherited from class ai.djl.training.dataset.RandomAccessDataset

getData, getData, randomSplit, size

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface ai.djl.repository.dataset.PreparedDataset

prepare

Method Detail

builder

public static ImageFolder.Builder builder()

Creates a new builder to build a ImageFolder.

Returns:

a new builder

getImagePath

protected java.nio.file.Path getImagePath(java.lang.String key)

Specified by:

getImagePath in class AbstractImageFolder

prepare

public void prepare(ai.djl.util.Progress progress)