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 |
---|
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.
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.
tensorcv.predicts.predictions.
PredictionScalar
(prediction_scalar_tensors, print_prefix)[source]¶Bases: tensorcv.predicts.predictions.PredictionBase
tensorcv.predicts.predictions.
PredictionMat
(prediction_tensors, save_prefix)[source]¶Bases: tensorcv.predicts.predictions.PredictionBase
tensorcv.predicts.predictions.
PredictionMeanScalar
(prediction_scalar_tensors, print_prefix)[source]¶tensorcv.predicts.simple.
SimpleFeedPredictor
(config)[source]¶Bases: tensorcv.predicts.base.Predictor
predictor with feed input