Package ai.djl.basicdataset.cv

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

    • Field Summary

      • 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

      All Methods Static Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      protected long availableSize()
      static BananaDetection.Builder builder()
      Creates a new builder to build a BananaDetection.
      java.util.List<java.lang.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.
      java.util.Optional<java.lang.Integer> getImageHeight()
      Returns the height of the images in the dataset.
      java.util.Optional<java.lang.Integer> getImageWidth()
      Returns the width of the images in the dataset.
      ai.djl.util.PairList<java.lang.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.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 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

      • getObjects

        public ai.djl.util.PairList<java.lang.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 java.util.List<java.lang.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 java.io.IOException,
                    ai.djl.translate.TranslateException
        Throws:
        java.io.IOException
        ai.djl.translate.TranslateException

      • getImage

        protected ai.djl.modality.cv.Image getImage​(long index)
                                     throws java.io.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:
        java.io.IOException - if the image could not be loaded

      • getImageWidth

        public java.util.Optional<java.lang.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 java.util.Optional<java.lang.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