ai.djl.basicdataset

Class AbstractImageFolder.ImageFolderBuilder<T extends AbstractImageFolder.ImageFolderBuilder>

java.lang.Object

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

ai.djl.basicdataset.AbstractImageFolder.ImageFolderBuilder<T>

Type Parameters:

T - the builder type

Direct Known Subclasses:

ImageFolder.Builder, ImageNet.Builder

Enclosing class:

AbstractImageFolder

public abstract static class AbstractImageFolder.ImageFolderBuilder<T extends AbstractImageFolder.ImageFolderBuilder>
extends ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<T>

Used to build an AbstractImageFolder.

Field Summary

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

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

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	ImageFolderBuilder()

Method Summary

Modifier and Type	Method and Description
T	optFlag(ai.djl.modality.cv.util.NDImageUtils.Flag flag) Sets the optional color mode flag.
T	setRepository(ai.djl.repository.Repository repository) Sets the repository containing the image folder.

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

getSampler, optBatchier, optDevice, optExecutor, optLimit, optPipeline, optTargetPipeline, self, setSampling, setSampling, setSampling

Methods inherited from class java.lang.Object

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

Constructor Detail

ImageFolderBuilder

protected ImageFolderBuilder()

Method Detail

optFlag

public T optFlag(ai.djl.modality.cv.util.NDImageUtils.Flag flag)

Sets the optional color mode flag.

Parameters:

flag - the color mode flag

Returns:

this builder

setRepository

public T setRepository(ai.djl.repository.Repository repository)

Sets the repository containing the image folder.

Parameters:

repository - the repository containing the image folder

Returns:

this builder