tensorcv.predicts package

Submodules

tensorcv.predicts.base module

class tensorcv.predicts.base.Predictor(config)[source]

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)[source]

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()[source]
run_predict()[source]

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)[source]

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)[source]

Args:

callbacks

tensorcv.predicts.predictions module

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

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)[source]
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)[source]

Bases: tensorcv.predicts.predictions.PredictionBase

__init__(prediction_scalar_tensors, print_prefix)[source]
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)[source]

Bases: tensorcv.predicts.predictions.PredictionBase

class tensorcv.predicts.predictions.PredictionMeanScalar(prediction_scalar_tensors, print_prefix)[source]

Bases: tensorcv.predicts.predictions.PredictionScalar

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

Bases: tensorcv.predicts.predictions.PredictionImage

tensorcv.predicts.simple module

class tensorcv.predicts.simple.SimpleFeedPredictor(config)[source]

Bases: tensorcv.predicts.base.Predictor

predictor with feed input

Module contents