ai.djl.basicdataset.cv

Class ImageDataset.BaseBuilder<T extends ImageDataset.BaseBuilder<T>>

java.lang.Object

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

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

Type Parameters:

T - the builder type

Direct Known Subclasses:

AbstractImageFolder.ImageFolderBuilder, BananaDetection.Builder, CocoDetection.Builder, PikachuDetection.Builder

Enclosing class:

ImageDataset

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

Used to build an ImageClassificationDataset.

Field Summary

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

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

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	BaseBuilder()

Method Summary

Modifier and Type	Method and Description
T	optFlag(ai.djl.modality.cv.Image.Flag flag) Sets the optional color mode flag.

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

addTargetTransform, addTransform, getSampler, optDataBatchifier, optDevice, optLabelBatchifier, optLimit, optPipeline, optPrefetchNumber, 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

BaseBuilder

protected BaseBuilder()

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