tensorcv.predicts package

Submodules

tensorcv.predicts.base module

class tensorcv.predicts.base.Predictor(config)

Bases: object

Base class for a predictor. Used to run all predictions.

config

PridectConfig – the config used for this predictor

model

ModelDes

input

DataFlow

sess

tf.Session

hooked_sess

tf.train.MonitoredSession

__init__(config)

Inits Predictor with config (PridectConfig).

Will create session as well as monitored sessions for each predictions, and load pre-trained parameters.

Parameters:config (PridectConfig) – the config used for this predictor

after_prediction()

run_predict()

Run predictions and the process after finishing predictions.

tensorcv.predicts.config module

class tensorcv.predicts.config.PridectConfig(dataflow=None, model=None, model_dir=None, model_name='', restore_vars=None, session_creator=None, predictions=None, batch_size=1, default_dirs=None)

Bases: object

__init__(dataflow=None, model=None, model_dir=None, model_name='', restore_vars=None, session_creator=None, predictions=None, batch_size=1, default_dirs=None)

Args:

callbacks

tensorcv.predicts.predictions module

class tensorcv.predicts.predictions.PredictionImage(prediction_image_tensors, save_prefix, merge_im=False, tanh=False, color=False)

Bases: tensorcv.predicts.predictions.PredictionBase

Predict image output and save as files.

Images are saved every batch. Each batch result can be save in one image or individule images.

__init__(prediction_image_tensors, save_prefix, merge_im=False, tanh=False, color=False)

Parameters:

• prediction_image_tensors (list) – a list of tensor names

• save_prefix (list) – a list of file prefix for saving each tensor in prediction_image_tensors

• merge_im (bool) – merge output of one batch or not

class tensorcv.predicts.predictions.PredictionScalar(prediction_scalar_tensors, print_prefix)

Bases: tensorcv.predicts.predictions.PredictionBase

__init__(prediction_scalar_tensors, print_prefix)

Parameters:

• prediction_scalar_tensors (list) – a list of tensor names

• print_prefix (list) – a list of name prefix for printing each tensor in prediction_scalar_tensors

class tensorcv.predicts.predictions.PredictionMat(prediction_tensors, save_prefix)

Bases: tensorcv.predicts.predictions.PredictionBase

class tensorcv.predicts.predictions.PredictionMeanScalar(prediction_scalar_tensors, print_prefix)

Bases: tensorcv.predicts.predictions.PredictionScalar

class tensorcv.predicts.predictions.PredictionOverlay(prediction_image_tensors, save_prefix, merge_im=False, tanh=False, color=False)

Bases: tensorcv.predicts.predictions.PredictionImage

tensorcv.predicts.simple module

class tensorcv.predicts.simple.SimpleFeedPredictor(config)

Bases: tensorcv.predicts.base.Predictor

predictor with feed input

Module contents