ai.djl.basicdataset

Class BananaDetection

java.lang.Object

ai.djl.training.dataset.RandomAccessDataset

ai.djl.basicdataset.BananaDetection

All Implemented Interfaces:

ai.djl.training.dataset.Dataset

public class BananaDetection
extends ai.djl.training.dataset.RandomAccessDataset

Banana image detection dataset contains a 3 x 256 x 256 image in the dataset with a banana of varying sizes in each image. There are 1000 training images and 100 testing images.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	BananaDetection.Builder A builder for a BananaDetection.

Nested classes/interfaces inherited from class ai.djl.training.dataset.RandomAccessDataset

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

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.training.dataset.RandomAccessDataset

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

Constructor Summary

Constructors

Constructor and Description
BananaDetection(BananaDetection.Builder builder) Creates a new instance of RandomAccessDataset with the given necessary configurations.

Method Summary

Modifier and Type	Method and Description
protected long	availableSize()
static BananaDetection.Builder	builder() Creates a new builder to build a BananaDetection.
protected ai.djl.training.dataset.Record	get(ai.djl.ndarray.NDManager manager, long index)
void	prepare(ai.djl.util.Progress progress)

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

getData, getData, randomSplit, size

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

Constructor Detail

BananaDetection

public BananaDetection(BananaDetection.Builder builder)

Creates a new instance of RandomAccessDataset with the given necessary configurations.

Parameters:

builder - a builder with the necessary configurations

Method Detail

builder

public static BananaDetection.Builder builder()

Creates a new builder to build a BananaDetection.

Returns:

a new builder

get

protected ai.djl.training.dataset.Record get(ai.djl.ndarray.NDManager manager,
                                             long index)
                                      throws java.io.IOException

Specified by:

get in class ai.djl.training.dataset.RandomAccessDataset

Throws:

java.io.IOException

availableSize

protected long availableSize()

Specified by:

availableSize in class ai.djl.training.dataset.RandomAccessDataset

prepare

public void prepare(ai.djl.util.Progress progress)
             throws java.io.IOException,
                    ai.djl.translate.TranslateException

Throws:

java.io.IOException

ai.djl.translate.TranslateException