Package ai.djl.basicdataset.cv.classification

Class CaptchaDataset

java.lang.Object
ai.djl.training.dataset.RandomAccessDataset
ai.djl.basicdataset.cv.classification.CaptchaDataset
All Implemented Interfaces:
ai.djl.training.dataset.Dataset
public class CaptchaDataset extends ai.djl.training.dataset.RandomAccessDataset
A Dataset featuring captcha images.

Each image is a 160x60 grayscale image featuring 5 or 6 digits where each digit ranges from 0-10. The dataset therefore features 6 labels. Each label ranges from 0-11 where 0-10 represent a recognized digit and 11 indicates that the value is not a digit (size 5 and not 6).

  • Nested Class Summary

    Nested Classes
    Modifier and Type
    Class
    Description
    static final class 
    CaptchaDataset.Builder
    A builder for a CaptchaDataset.

    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<T>>

    Nested classes/interfaces inherited from interface ai.djl.training.dataset.Dataset

    ai.djl.training.dataset.Dataset.Usage

  • Field Summary

    Fields
    Modifier and Type
    Field
    Description
    static final int
    CAPTCHA_LENGTH
     
    static final int
    CAPTCHA_OPTIONS
     
    static final int
    IMAGE_HEIGHT
     
    static final int
    IMAGE_WIDTH
     

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

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

  • Constructor Summary

    Constructors
    Constructor
    Description
    CaptchaDataset(CaptchaDataset.Builder builder)
    Creates a new instance of CaptchaDataset.

  • Method Summary

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

  • Field Details

  • Constructor Details

    • CaptchaDataset

      public CaptchaDataset(CaptchaDataset.Builder builder)
      Creates a new instance of CaptchaDataset.
      Parameters:
      builder - a builder with the necessary configurations

  • Method Details

    • builder

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

    • get

      public ai.djl.training.dataset.Record get(ai.djl.ndarray.NDManager manager, long index) throws IOException
      Specified by:
      get in class ai.djl.training.dataset.RandomAccessDataset
      Throws:
      IOException

    • 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
      Throws:
      IOException