ch.epfl.leb.defcon.predictors.internal

Class DefaultPredictor

java.lang.Object

ch.epfl.leb.defcon.predictors.internal.AbstractPredictor

ch.epfl.leb.defcon.predictors.internal.DefaultPredictor

All Implemented Interfaces:

Predictor

public class DefaultPredictor
extends AbstractPredictor
implements Predictor

Makes density map predictions from images.

Author:

Baptiste Ottino, Kyle M. Douglass

Field Summary

Fields inherited from class ch.epfl.leb.defcon.predictors.internal.AbstractPredictor

isClosed, tfSession

Constructor Summary

Constructors

Constructor and Description
DefaultPredictor()

Method Summary

Modifier and Type	Method and Description
protected void	finalize() Failsafe in case the TensorFlow session has not been closed.
double	getCount() Returns the most recent count.
FloatProcessor	getDensityMap() Returns the most recently calculated density map prediction.
FloatProcessor	getLocalCountMap() Returns the most recently-calculated local count map.
double	getMaximumLocalCount(int boxSize) Returns the maximum local count value.
void	predict(ImageProcessor ip) Makes a density map prediction from a 2D image.

Methods inherited from class ch.epfl.leb.defcon.predictors.internal.AbstractPredictor

close, imageToTensor, setup

Methods inherited from class java.lang.Object

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

Methods inherited from interface ch.epfl.leb.defcon.predictors.Predictor

close, setup

Constructor Detail

DefaultPredictor

public DefaultPredictor()

Method Detail

getCount

public double getCount()
                throws UninitializedPredictorException

Returns the most recent count.

Specified by:

getCount in interface Predictor

Returns:

The predicted count from the density map.

Throws:

UninitializedPredictorException

getDensityMap

public FloatProcessor getDensityMap()
                             throws UninitializedPredictorException

Returns the most recently calculated density map prediction.

Specified by:

getDensityMap in interface Predictor

Returns:

The predicted density map.

Throws:

UninitializedPredictorException

getLocalCountMap

public FloatProcessor getLocalCountMap()
                                throws NoLocalCountMapException

Returns the most recently-calculated local count map.

Specified by:

getLocalCountMap in interface Predictor

Returns:

The most recently calculated local count map.

Throws:

NoLocalCountMapException

getMaximumLocalCount

public double getMaximumLocalCount(int boxSize)
                            throws UninitializedPredictorException

Returns the maximum local count value. This value is obtained by convolving the density map with a square kernel whose values are all 1 and then taking the maximum of the resulting map. It effectively produces the highest count value over length scales equal to the size of the kernel.

Specified by:

getMaximumLocalCount in interface Predictor

Parameters:

boxSize - The width of the square kernel.

Returns:

The maximum local count from the density map.

Throws:

UninitializedPredictorException

predict

public void predict(ImageProcessor ip)
             throws ImageBitDepthException,
                    SessionClosedException

Makes a density map prediction from a 2D image. If either of the image's width or height is not divisible by 4, they will be cropped to the next largest multiple of four.

Specified by:

predict in interface Predictor

Parameters:

ip - The image to perform a prediction on.

Throws:

ImageBitDepthException

SessionClosedException

finalize

protected void finalize()
                 throws Throwable

Failsafe in case the TensorFlow session has not been closed.

Overrides:

finalize in class AbstractPredictor

Throws:

Throwable