ScenePredictor

class ScenePredictor

Bases: object

Class for making predictions on a scen using a model-bundle.

__init__(model_bundle_uri: str, predict_options: str | dict | rastervision.core.rv_pipeline.rv_pipeline_config.PredictOptions | None = None, tmp_dir: Optional[str] = None)

Creates a new Predictor.

Parameters

• model_bundle_uri (str) – URI of the model bundle to use. Can be any type of URI that Raster Vision can read.

• predict_options (str | dict | rastervision.core.rv_pipeline.rv_pipeline_config.PredictOptions | None) – Either a URI to a serialized PredictOptions or a dict representing a serialized PredictOptions or a PredictOptions instance.

• tmp_dir (Optional[str]) – Temporary directory in which to store files that are used by the Predictor.

Methods

__init__(model_bundle_uri[, ...])	Creates a new Predictor.
build_scene(scene_config)
predict(scene_config)	Generate predictions for the given image.
predict_scene(scene)	Generate predictions for the given scene.

__init__(model_bundle_uri: str, predict_options: str | dict | rastervision.core.rv_pipeline.rv_pipeline_config.PredictOptions | None = None, tmp_dir: Optional[str] = None)

Creates a new Predictor.

Parameters

• model_bundle_uri (str) – URI of the model bundle to use. Can be any type of URI that Raster Vision can read.

• predict_options (str | dict | rastervision.core.rv_pipeline.rv_pipeline_config.PredictOptions | None) – Either a URI to a serialized PredictOptions or a dict representing a serialized PredictOptions or a PredictOptions instance.

• tmp_dir (Optional[str]) – Temporary directory in which to store files that are used by the Predictor.

build_scene(scene_config: str | dict | SceneConfig) → Scene

Parameters

scene_config (str | dict | SceneConfig) –

Return type

Scene

predict(scene_config: str | dict | rastervision.core.data.scene_config.SceneConfig) → None

Generate predictions for the given image.

Parameters

• scene_config_uri – URI to a serialized ScenConfig.

• scene_config (str | dict | rastervision.core.data.scene_config.SceneConfig) –

Return type

None

predict_scene(scene: Scene) → None

Generate predictions for the given scene.

Parameters

scene (Scene) – Scene to predict on.

Return type

None