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

dataBatchifier, device, executor, labelBatchifier, 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.Image.Flag flag) Sets the optional color mode flag.
T	optMaxDepth(int maxDepth) Sets the depth of the image folder.
T	setRepository(ai.djl.repository.Repository repository) Sets the repository containing the image folder.
T	setRepositoryPath(java.nio.file.Path path) Sets the repository file path containing the image folder.
T	setRepositoryPath(java.lang.String path) Sets the repository file path containing the image folder.

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

addTargetTransform, addTransform, getSampler, optDataBatchifier, optDevice, optExecutor, optLabelBatchifier, 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.Image.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

setRepositoryPath

public T setRepositoryPath(java.lang.String path)

Sets the repository file path containing the image folder.

Parameters:

path - the repository file path containing the image folder

Returns:

this builder

setRepositoryPath

public T setRepositoryPath(java.nio.file.Path path)

Sets the repository file path containing the image folder.

Parameters:

path - the repository file path containing the image folder

Returns:

this builder

optMaxDepth

public T optMaxDepth(int maxDepth)

Sets the depth of the image folder.

Parameters:

maxDepth - the maximum number of directory levels to visit

Returns:

this builder