Package ai.djl.basicdataset.cv.classification

Class ImageFolder.Builder

java.lang.Object

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

ai.djl.basicdataset.cv.ImageDataset.BaseBuilder<T>

ai.djl.basicdataset.cv.classification.AbstractImageFolder.ImageFolderBuilder<ImageFolder.Builder>

ai.djl.basicdataset.cv.classification.ImageFolder.Builder

Enclosing class:

ImageFolder

public static final class ImageFolder.Builder
extends AbstractImageFolder.ImageFolderBuilder<ImageFolder.Builder>

A builder for the ImageFolder.

Field Summary

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

dataBatchifier, device, labelBatchifier, limit, pipeline, prefetchNumber, sampler, targetPipeline

Method Summary

Modifier and Type	Method	Description
ImageFolder	build()	Builds the ImageFolder.
protected ImageFolder.Builder	self()

Methods inherited from class ai.djl.basicdataset.cv.classification.AbstractImageFolder.ImageFolderBuilder

optImageHeight, optImageSize, optImageWidth, optMaxDepth, setRepository, setRepositoryPath, setRepositoryPath

Methods inherited from class ai.djl.basicdataset.cv.ImageDataset.BaseBuilder

optFlag

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

addTargetTransform, addTransform, getSampler, optDataBatchifier, optDevice, optLabelBatchifier, optLimit, optPipeline, optPrefetchNumber, optTargetPipeline, setSampling, setSampling, setSampling

Methods inherited from class java.lang.Object

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

Method Detail

self

protected ImageFolder.Builder self()

Specified by:

self in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<ImageFolder.Builder>

build

public ImageFolder build()

Builds the ImageFolder.

Returns:

the ImageFolder