Package 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 anImageClassificationDataset
.
-
-
Constructor Summary
Constructors Modifier Constructor Description protected
BaseBuilder()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description T
optFlag(ai.djl.modality.cv.Image.Flag flag)
Sets the optional color mode flag.
-
-
-
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
-
-