.. java:import:: ch.epfl.leb.defcon.utils GraphBuilder

.. java:import:: ij ImagePlus

.. java:import:: net.imagej.tensorflow Tensors

.. java:import:: net.imglib2.img Img

.. java:import:: net.imglib2.img.display.imagej ImageJFunctions

.. java:import:: net.imglib2.type.numeric.real FloatType

.. java:import:: org.tensorflow Graph

.. java:import:: org.tensorflow Output

.. java:import:: org.tensorflow SavedModelBundle

.. java:import:: org.tensorflow Session

.. java:import:: org.tensorflow Tensor

AbstractPredictor
=================

.. java:package:: ch.epfl.leb.defcon.predictors.internal
   :noindex:

.. java:type:: public abstract class AbstractPredictor

   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

Fields
------
isClosed
^^^^^^^^

.. java:field:: protected boolean isClosed
   :outertype: AbstractPredictor

   Has the TensorFlow session been closed?

tfSession
^^^^^^^^^

.. java:field:: protected Session tfSession
   :outertype: AbstractPredictor

   A copy of the current TensorFlow session.

Methods
-------
close
^^^^^

.. java:method:: public void close()
   :outertype: AbstractPredictor

   Closes resources associated with this predictor.

finalize
^^^^^^^^

.. java:method:: @Override protected void finalize() throws Throwable
   :outertype: AbstractPredictor

   Failsafe in case the TensorFlow session has not been closed.

   :throws java.lang.Throwable:

imageToTensor
^^^^^^^^^^^^^

.. java:method:: protected static Tensor<Float> imageToTensor(ImagePlus imp)
   :outertype: AbstractPredictor

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

   :param imp: The image to convert.
   :return: A tensor representing the data in the original image.

setup
^^^^^

.. java:method:: public void setup(String pathToModel)
   :outertype: AbstractPredictor

   Initializes the predictor.

   :param pathToModel: The path to a saved TensorFlow model bundle.