boofcv.deepboof

Class BaseImageClassifier

java.lang.Object

boofcv.deepboof.BaseImageClassifier

All Implemented Interfaces:

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

Direct Known Subclasses:

ImageClassifierNiNImageNet, ImageClassifierResNet, ImageClassifierVggCifar10

public abstract class BaseImageClassifier
extends java.lang.Object
implements ImageClassifier<boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>>

Base class for ImageClassifiers which implements common elements

Nested Class Summary

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

ImageClassifier.Score

Field Summary

Fields

Modifier and Type	Field and Description
protected java.util.List<java.lang.String>	categories
protected int	categoryBest
protected org.ddogleg.struct.FastQueue<ImageClassifier.Score>	categoryScores
protected boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>	imageRgb
protected int	imageSize
protected boofcv.struct.image.ImageType<boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>>	imageType
protected ClipAndReduce<boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>>	massage
protected deepboof.graph.FunctionSequence<deepboof.tensors.Tensor_F32,deepboof.Function<deepboof.tensors.Tensor_F32>>	network
protected deepboof.tensors.Tensor_F32	tensorInput
protected deepboof.tensors.Tensor_F32	tensorOutput

Constructor Summary

Constructors

Constructor and Description
BaseImageClassifier(int imageSize)

Method Summary

Modifier and Type	Method and Description
void	classify(boofcv.struct.image.Planar<boofcv.struct.image.GrayF32> image) The original implementation takes in an image then crops it randomly.
java.util.List<ImageClassifier.Score>	getAllResults() Returns a list of all the likely categories for the image.
int	getBestResult() Returns the category which was the best fit.
java.util.List<java.lang.String>	getCategories() Returns a list of all the possible categories that a scene can be classified as
boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>	getImageRgb()
boofcv.struct.image.ImageType<boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>>	getInputType() Returns the type of input image
protected void	innerProcess(deepboof.tensors.Tensor_F32 tensorInput)
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 java.lang.Object

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

Methods inherited from interface boofcv.abst.scene.ImageModelBase

loadModel

Field Detail

network

protected deepboof.graph.FunctionSequence<deepboof.tensors.Tensor_F32,deepboof.Function<deepboof.tensors.Tensor_F32>> network

categories

protected java.util.List<java.lang.String> categories

imageType

protected boofcv.struct.image.ImageType<boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>> imageType

massage

protected ClipAndReduce<boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>> massage

imageSize

protected int imageSize

imageRgb

protected boofcv.struct.image.Planar<boofcv.struct.image.GrayF32> imageRgb

tensorInput

protected deepboof.tensors.Tensor_F32 tensorInput

tensorOutput

protected deepboof.tensors.Tensor_F32 tensorOutput

categoryScores

protected org.ddogleg.struct.FastQueue<ImageClassifier.Score> categoryScores

categoryBest

protected int categoryBest

Constructor Detail

BaseImageClassifier

public BaseImageClassifier(int imageSize)

Method Detail

getInputType

public boofcv.struct.image.ImageType<boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>> getInputType()

Description copied from interface: ImageModelBase

Returns the type of input image

Specified by:

getInputType in interface ImageModelBase<boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>>

Returns:

input image type

classify

public void classify(boofcv.struct.image.Planar<boofcv.struct.image.GrayF32> image)

The original implementation takes in an image then crops it randomly. This is primarily for training but is replicated here to reduce the number of differences

Specified by:

classify in interface ImageClassifier<boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>>

Parameters:

image - Image being processed. Must be RGB image. Pixel values must have values from 0 to 255.

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

innerProcess

protected void innerProcess(deepboof.tensors.Tensor_F32 tensorInput)

getBestResult

public int getBestResult()

Description copied from interface: ImageClassifier

Returns the category which was the best fit.

Specified by:

getBestResult in interface ImageClassifier<boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>>

Returns:

best fit category

getAllResults

public java.util.List<ImageClassifier.Score> getAllResults()

Description copied from interface: ImageClassifier

Returns a list of all the likely categories for the image. What is likely is implementation dependent. Each category should be included at least once and might not be included at all.

Specified by:

getAllResults in interface ImageClassifier<boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>>

Returns:

List of categories and scores

getCategories

public java.util.List<java.lang.String> getCategories()

Description copied from interface: ImageClassifier

Returns a list of all the possible categories that a scene can be classified as

Specified by:

getCategories in interface ImageClassifier<boofcv.struct.image.Planar<boofcv.struct.image.GrayF32>>

Returns:

Names of categories

getImageRgb

public boofcv.struct.image.Planar<boofcv.struct.image.GrayF32> getImageRgb()