ch.epfl.leb.defcon.predictors

Interface Predictor

All Known Implementing Classes:

DefaultPredictor

public interface Predictor

Makes density map predictions from images.

Author:

Kyle M. Douglass

Method Summary

Modifier and Type	Method and Description
void	close() Closes resources associated with this predictor.
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.
void	setup(String pathToModel) Initializes the predictor with a saved TensorFlow model bundle.

Method Detail

close

void close()

Closes resources associated with this predictor.

getCount

double getCount()
         throws UninitializedPredictorException

Returns the most recent count.

Returns:

The predicted count from the density map.

Throws:

UninitializedPredictorException

getDensityMap

FloatProcessor getDensityMap()
                      throws UninitializedPredictorException

Returns the most recently calculated density map prediction.

Returns:

The predicted density map.

Throws:

UninitializedPredictorException

getLocalCountMap

FloatProcessor getLocalCountMap()
                         throws NoLocalCountMapException

Returns the most recently-calculated local count map.

Returns:

The most recently calculated local count map.

Throws:

NoLocalCountMapException

getMaximumLocalCount

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.

Parameters:

boxSize - The width of the square kernel.

Returns:

The maximum local count from the density map.

Throws:

UninitializedPredictorException

predict

void predict(ImageProcessor ip)
      throws ImageBitDepthException,
             SessionClosedException

Makes a density map prediction from a 2D image.

Parameters:

ip - The image to perform a prediction on.

Throws:

ImageBitDepthException

SessionClosedException

setup

void setup(String pathToModel)

Initializes the predictor with a saved TensorFlow model bundle.

Parameters:

pathToModel - The path to a saved TensorFlow model bundle.