Class ImageNet

  • All Implemented Interfaces:
    ai.djl.training.dataset.Dataset

    public class ImageNet
    extends AbstractImageFolder
    ImageNet is an image classification dataset from http://image-net.org 2012 Classification dataset.

    Each image might have different Shapes.

    • Method Detail

      • getWordNetIds

        public java.lang.String[] getWordNetIds()
        Returns all WordNet ids of this ImageNet dataset.
        Returns:
        all WordNet ids of this ImageNet dataset
      • getClassNames

        public java.lang.String[] getClassNames()
        Returns all class names of this ImageNet dataset.
        Returns:
        all class names of this ImageNet dataset
      • getClassFull

        public java.lang.String[] getClassFull()
        Returns all full class names of this ImageNet dataset.
        Returns:
        all full class names of this ImageNet dataset
      • prepare

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