ai.djl.basicdataset

Class PikachuDetection

java.lang.Object

ai.djl.training.dataset.RandomAccessDataset

ai.djl.basicdataset.PikachuDetection

All Implemented Interfaces:

ai.djl.training.dataset.Dataset

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

Pikachu image detection dataset that contains multiple Pikachus in each image.

Nested Class Summary

Nested Classes

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

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

Modifier	Constructor and Description
protected	PikachuDetection(PikachuDetection.Builder builder)

Method Summary

Modifier and Type	Method and Description
protected long	availableSize()
static PikachuDetection.Builder	builder() Creates a new builder to build a PikachuDetection.
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

PikachuDetection

protected PikachuDetection(PikachuDetection.Builder builder)

Method Detail

builder

public static PikachuDetection.Builder builder()

Creates a new builder to build a PikachuDetection.

Returns:

a new builder

prepare

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

Throws:

java.io.IOException

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