boofcv.deepboof

Class ImageClassifierNiNImageNet

java.lang.Object

boofcv.deepboof.BaseImageClassifier

boofcv.deepboof.ImageClassifierNiNImageNet

All Implemented Interfaces:

ImageClassifier<boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>>, ImageModelBase<boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>>

public class ImageClassifierNiNImageNet
extends BaseImageClassifier

Pretrained Network-in-Network (NiN) image classifier using imagenet data. Trained by szagoruyko [1,2] and achieves 62.6% top1 center crop accuracy on validation set.

[1] https://gist.github.com/szagoruyko/0f5b4c5e2d2b18472854
[2] https://github.com/soumith/imagenet-multiGPU.torch/blob/master/models/ninbn.lua

Nested Class Summary

Nested classes/interfaces inherited from interface boofcv.abst.scene.ImageClassifier

ImageClassifier.Score

Field Summary

Fields inherited from class boofcv.deepboof.BaseImageClassifier

categories, categoryBest, categoryScores, imageRgb, imageSize, imageType, massage, network, tensorInput, tensorOutput

Constructor Summary

Constructors

Constructor and Description
ImageClassifierNiNImageNet()

Method Summary

Modifier and Type	Method and Description
void	loadModel(java.io.File directory) Loads the model at the specified location.
protected boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>	preprocess(boofcv.struct.image.Planar<boofcv.struct.image.GrayF32> image) Massage the input image into a format recognized by the network

Methods inherited from class boofcv.deepboof.BaseImageClassifier

classify, getAllResults, getBestResult, getCategories, getImageRgb, getInputType, innerProcess

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

ImageClassifierNiNImageNet

public ImageClassifierNiNImageNet()

Method Detail

loadModel

public void loadModel(java.io.File directory)
               throws java.io.IOException

Description copied from interface: ImageModelBase

Loads the model at the specified location. See documentation of the classifier for what needs to be passed in here.

Parameters:

directory - Path to directory or file containing the model

Throws:

java.io.IOException

preprocess

protected boofcv.struct.image.Planar<boofcv.struct.image.GrayF32> preprocess(boofcv.struct.image.Planar<boofcv.struct.image.GrayF32> image)

Massage the input image into a format recognized by the network

Overrides:

preprocess in class BaseImageClassifier