Package ai.djl.basicdataset.cv

Class BananaDetection

java.lang.Object
ai.djl.training.dataset.RandomAccessDataset
ai.djl.basicdataset.cv.ImageDataset
ai.djl.basicdataset.cv.ObjectDetectionDataset
ai.djl.basicdataset.cv.BananaDetection
All Implemented Interfaces:
ai.djl.training.dataset.Dataset
public class BananaDetection extends ObjectDetectionDataset
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
    Description
    static final class 
    BananaDetection.Builder
    A builder for a BananaDetection.

    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.ImageDataset

    flag

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

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

  • Constructor Summary

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

  • Method Summary

    Modifier and Type
    Method
    Description
    protected long
    availableSize()
    static BananaDetection.Builder
    builder()
    Creates a new builder to build a BananaDetection.
    List<String>
    getClasses()
    Returns the classes that detected objects in the dataset can be classified into.
    protected ai.djl.modality.cv.Image
    getImage(long index)
    Returns the image at the given index in the dataset.
    Optional<Integer>
    getImageHeight()
    Returns the height of the images in the dataset.
    Optional<Integer>
    getImageWidth()
    Returns the width of the images in the dataset.
    ai.djl.util.PairList<Long,ai.djl.modality.cv.output.Rectangle>
    getObjects(long index)
    Returns the list of objects in the image at the given index.
    void
    prepare(ai.djl.util.Progress progress)

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

    get

    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

    matchingTranslatorOptions, prepare

  • Constructor Details

    • 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 Details

    • builder

      public static BananaDetection.Builder builder()
      Creates a new builder to build a BananaDetection.
      Returns:
      a new builder

    • getObjects

      public ai.djl.util.PairList<Long,ai.djl.modality.cv.output.Rectangle> getObjects(long index)
      Returns the list of objects in the image at the given index.
      Specified by:
      getObjects in class ObjectDetectionDataset
      Parameters:
      index - the index (if the dataset is a list of data items)
      Returns:
      the list of objects in the image. The long is the class number of the index into the list of classes of the desired class name. The rectangle is the location of the object inside the image.

    • getClasses

      public List<String> getClasses()
      Returns the classes that detected objects in the dataset can be classified into.
      Specified by:
      getClasses in class ObjectDetectionDataset
      Returns:
      the classes that detected objects in the dataset can be classified into.

    • availableSize

      protected long availableSize()
      Specified by:
      availableSize in class ai.djl.training.dataset.RandomAccessDataset

    • prepare

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

    • getImage

      protected ai.djl.modality.cv.Image getImage(long index) throws IOException
      Returns the image at the given index in the dataset.
      Specified by:
      getImage in class ImageDataset
      Parameters:
      index - the index (if the dataset is a list of data items)
      Returns:
      the image
      Throws:
      IOException - if the image could not be loaded

    • getImageWidth

      public Optional<Integer> getImageWidth()
      Returns the width of the images in the dataset.
      Specified by:
      getImageWidth in class ImageDataset
      Returns:
      the width of the images in the dataset

    • getImageHeight

      public Optional<Integer> getImageHeight()
      Returns the height of the images in the dataset.
      Specified by:
      getImageHeight in class ImageDataset
      Returns:
      the height of the images in the dataset