Package ai.djl.basicdataset.cv.classification

Class ImageNet.Builder

java.lang.Object

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

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

ai.djl.basicdataset.cv.classification.AbstractImageFolder.ImageFolderBuilder<ImageNet.Builder>

ai.djl.basicdataset.cv.classification.ImageNet.Builder

Enclosing class:

ImageNet

public static class ImageNet.Builder
extends AbstractImageFolder.ImageFolderBuilder<ImageNet.Builder>

A builder for a ImageNet.

Field Summary

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

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

Method Summary

Modifier and Type	Method	Description
ImageNet	build()	Builds the ImageNet.
ImageNet.Builder	optUsage(ai.djl.training.dataset.Dataset.Usage usage)	Sets the optional usage.
ImageNet.Builder	self()

Methods inherited from class ai.djl.basicdataset.cv.classification.AbstractImageFolder.ImageFolderBuilder

optImageHeight, optImageSize, optImageWidth, optMaxDepth, setRepository, setRepositoryPath, setRepositoryPath

Methods inherited from class ai.djl.basicdataset.cv.ImageDataset.BaseBuilder

optFlag

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

addTargetTransform, addTransform, getSampler, optDataBatchifier, optDevice, optLabelBatchifier, optLimit, optPipeline, optPrefetchNumber, optTargetPipeline, setSampling, setSampling, setSampling

Methods inherited from class java.lang.Object

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

Method Details

optUsage

public ImageNet.Builder optUsage(ai.djl.training.dataset.Dataset.Usage usage)

Sets the optional usage.

Parameters:

usage - the usage

Returns:

this builder

self

public ImageNet.Builder self()

Specified by:

self in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<ImageNet.Builder>

build

public ImageNet build()

Builds the ImageNet.

Returns:

the ImageNet