Class ImageClassifierVggCifar10

java.lang.Object
boofcv.deepboof.BaseImageClassifier
boofcv.deepboof.ImageClassifierVggCifar10
All Implemented Interfaces:
ImageClassifier<Planar<GrayF32>>, ImageModelBase<Planar<GrayF32>>

public class ImageClassifierVggCifar10
extends BaseImageClassifier
Image classification using VGG network trained in CIFAR 10 data. On the CIFAR 10 training set it get has 89.9% accuracy. This dataset contains images in 10 categories and 32x32 images.
See Also:
szagoruyko/cifar.torch
  • Constructor Details

    • ImageClassifierVggCifar10

      public ImageClassifierVggCifar10()
  • Method Details

    • loadModel

      public void loadModel​(File directory) throws IOException
      Expects there to be two files in the provided directory:
      YuvStatistics.txt
      model.net
      Parameters:
      directory - Directory containing model files
      Throws:
      IOException - Throw if anything goes wrong while reading data
    • preprocess

      protected Planar<GrayF32> preprocess​(Planar<GrayF32> image)
      Description copied from class: BaseImageClassifier
      Massage the input image into a format recognized by the network
      Overrides:
      preprocess in class BaseImageClassifier