ch.epfl.leb.defcon.predictors.internal

Class AbstractPredictor

java.lang.Object

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

Direct Known Subclasses:

DefaultPredictor, MaxCountFCN

public abstract class AbstractPredictor
extends Object

A base implementation for DEFCoN predictors. A predictor is a class that takes an image(s) as input and estimates a density map of object locations from it.

Author:

Kyle M. Douglass

Field Summary

Fields

Modifier and Type	Field and Description
protected boolean	isClosed Has the TensorFlow session been closed?
protected org.tensorflow.Session	tfSession A copy of the current TensorFlow session.

Constructor Summary

Constructors

Constructor and Description
AbstractPredictor()

Method Summary

Modifier and Type	Method and Description
void	close() Closes resources associated with this predictor.
protected void	finalize() Failsafe in case the TensorFlow session has not been closed.
protected static org.tensorflow.Tensor<Float>	imageToTensor(ImagePlus imp) Converts an ImageJ image to a TensorFlow tensor.
void	setup(String pathToModel) Initializes the predictor.

Methods inherited from class java.lang.Object

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

Field Detail

tfSession

protected org.tensorflow.Session tfSession

A copy of the current TensorFlow session.

isClosed

protected boolean isClosed

Has the TensorFlow session been closed?

Constructor Detail

AbstractPredictor

public AbstractPredictor()

Method Detail

close

public void close()

Closes resources associated with this predictor.

setup

public void setup(String pathToModel)

Initializes the predictor.

Parameters:

pathToModel - The path to a saved TensorFlow model bundle.

imageToTensor

protected static org.tensorflow.Tensor<Float> imageToTensor(ImagePlus imp)

Converts an ImageJ image to a TensorFlow tensor. The original image is preserved.

Parameters:

imp - The image to convert.

Returns:

A tensor representing the data in the original image.

finalize

protected void finalize()
                 throws Throwable

Failsafe in case the TensorFlow session has not been closed.

Overrides:

finalize in class Object

Throws:

Throwable