Package ai.djl.basicdataset.cv.classification

Class FruitsFreshAndRotten

java.lang.Object

ai.djl.training.dataset.RandomAccessDataset

ai.djl.basicdataset.cv.ImageDataset

ai.djl.basicdataset.cv.classification.ImageClassificationDataset

ai.djl.basicdataset.cv.classification.AbstractImageFolder

ai.djl.basicdataset.cv.classification.FruitsFreshAndRotten

All Implemented Interfaces:

ai.djl.training.dataset.Dataset

public final class FruitsFreshAndRotten
extends AbstractImageFolder

FruitRottenFresh classification dataset that contains the same fruit where rotten and fresh class are stored in different sub folders.

 It is structured similar to ImageFolders as follows:
      root/freshapples/1.png
      root/freshapples/2.png
      ...
      root/rottenapples/1.png
      root/rottenapples/2.png
      ...
      root/freshbanana/1.png
      root/freshbanana/2.png
      ...
      root/rottenbanana/1.png
      root/rottenbanana/2.png
      ...
 

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static final class	FruitsFreshAndRotten.Builder	A builder for the FruitsFreshAndRotten.

Nested classes/interfaces inherited from class ai.djl.basicdataset.cv.classification.AbstractImageFolder

AbstractImageFolder.ImageFolderBuilder<T extends AbstractImageFolder.ImageFolderBuilder<T>>

Nested classes/interfaces inherited from class ai.djl.basicdataset.cv.ImageDataset

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

Nested classes/interfaces inherited from interface ai.djl.training.dataset.Dataset

ai.djl.training.dataset.Dataset.Usage

Field Summary

Fields inherited from class ai.djl.basicdataset.cv.classification.AbstractImageFolder

items, synset

Fields inherited from class ai.djl.basicdataset.cv.ImageDataset

flag

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

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

Method Summary

Modifier and Type	Method	Description
static FruitsFreshAndRotten.Builder	builder()	Creates a new builder to build a FruitsFreshAndRotten.
protected Path	getImagePath(String key)
void	prepare(ai.djl.util.Progress progress)

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

availableSize, getClasses, getClassNumber, getImage, getImageHeight, getImageWidth, getSynset, isImage, listImages

Methods inherited from class ai.djl.basicdataset.cv.classification.ImageClassificationDataset

get, matchingTranslatorOptions

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

getImageChannels, getRecordImage

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

getData, getData, getData, getData, newSubDataset, newSubDataset, randomSplit, size, subDataset, subDataset, subDataset, subDataset, toArray

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.training.dataset.Dataset

prepare

Method Details

builder

public static FruitsFreshAndRotten.Builder builder()

Creates a new builder to build a FruitsFreshAndRotten.

Returns:

a new builder

getImagePath

protected Path getImagePath(String key)

Specified by:

getImagePath in class AbstractImageFolder

prepare

public void prepare(ai.djl.util.Progress progress)
             throws IOException

Throws:

IOException