Interface ImageModelBase<T extends ImageBase<T>>

All Known Subinterfaces:
ImageCaptioning<T>, ImageClassifier<T>
All Known Implementing Classes:
BaseImageClassifier, ImageClassifierNiNImageNet, ImageClassifierResNet, ImageClassifierVggCifar10

public interface ImageModelBase<T extends ImageBase<T>>
Base class for algorithms which process an image and load a model to do so
  • Method Summary

    Modifier and Type Method Description
    ImageType<T> getInputType()
    Returns the type of input image
    void loadModel​(File path)
    Loads the model at the specified location.
  • Method Details

    • loadModel

      void loadModel​(File path) throws IOException
      Loads the model at the specified location. See documentation of the classifier for what needs to be passed in here.
      Parameters:
      path - Path to directory or file containing the model
      Throws:
      IOException
    • getInputType

      ImageType<T> getInputType()
      Returns the type of input image
      Returns:
      input image type