public class CocoDetection
extends ai.djl.training.dataset.RandomAccessDataset
implements ai.djl.repository.dataset.ZooDataset
Each image might have different Shapes.
| Modifier and Type | Class and Description |
|---|---|
static class |
CocoDetection.Builder
A builder to construct a
CocoDetection. |
| Modifier and Type | Method and Description |
|---|---|
protected long |
availableSize() |
static CocoDetection.Builder |
builder()
Creates a builder to build a
CocoDetection. |
ai.djl.training.dataset.Record |
get(ai.djl.ndarray.NDManager manager,
long index) |
ai.djl.repository.Artifact |
getArtifact() |
ai.djl.repository.MRL |
getMrl() |
ai.djl.repository.Repository |
getRepository() |
ai.djl.training.dataset.Dataset.Usage |
getUsage() |
boolean |
isPrepared() |
void |
prepareData(ai.djl.training.dataset.Dataset.Usage usage) |
void |
setPrepared(boolean prepared) |
void |
useDefaultArtifact() |
getData, getData, randomSplit, sizepublic static CocoDetection.Builder builder()
CocoDetection.public ai.djl.repository.MRL getMrl()
getMrl in interface ai.djl.repository.dataset.ZooDatasetpublic ai.djl.repository.Repository getRepository()
getRepository in interface ai.djl.repository.dataset.ZooDatasetpublic ai.djl.repository.Artifact getArtifact()
getArtifact in interface ai.djl.repository.dataset.ZooDatasetpublic ai.djl.training.dataset.Dataset.Usage getUsage()
getUsage in interface ai.djl.repository.dataset.ZooDatasetpublic boolean isPrepared()
isPrepared in interface ai.djl.repository.dataset.ZooDatasetpublic void setPrepared(boolean prepared)
setPrepared in interface ai.djl.repository.dataset.ZooDatasetpublic void useDefaultArtifact()
throws java.io.IOException
useDefaultArtifact in interface ai.djl.repository.dataset.ZooDatasetjava.io.IOExceptionpublic ai.djl.training.dataset.Record get(ai.djl.ndarray.NDManager manager,
long index)
throws java.io.IOException
get in class ai.djl.training.dataset.RandomAccessDatasetjava.io.IOExceptionpublic void prepareData(ai.djl.training.dataset.Dataset.Usage usage)
throws java.io.IOException
prepareData in interface ai.djl.repository.dataset.ZooDatasetjava.io.IOExceptionprotected long availableSize()
availableSize in class ai.djl.training.dataset.RandomAccessDataset