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 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
    Description
    protected
    BaseBuilder()
     

  • Method Summary

    Modifier and Type
    Method
    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 Details

    • BaseBuilder

      protected BaseBuilder()

  • Method Details

    • 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