ai.djl.basicdataset

Class CocoDetection

java.lang.Object

ai.djl.training.dataset.RandomAccessDataset

ai.djl.basicdataset.CocoDetection

All Implemented Interfaces:

ai.djl.training.dataset.Dataset

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

Coco image detection dataset from http://cocodataset.org/#home.

Each image might have different Shapes.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	CocoDetection.Builder A builder to construct a CocoDetection.

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

Method Summary

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

Method Detail

builder

public static CocoDetection.Builder builder()

Creates a builder to build a CocoDetection.

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

prepare

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

Throws:

java.io.IOException

availableSize

protected long availableSize()

Specified by:

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