Source code for rastervision.core.data.scene_config

from typing import TYPE_CHECKING, List, Optional

from rastervision.pipeline.config import (Config, ConfigError, register_config,
                                          Field)
from rastervision.core.data.raster_source import RasterSourceConfig
from rastervision.core.data.label_source import LabelSourceConfig
from rastervision.core.data.label_store import LabelStoreConfig
from rastervision.core.data.scene import Scene
from rastervision.core.data.utils import get_polygons_from_uris

if TYPE_CHECKING:
    from rastervision.core.data import ClassConfig
    from rastervision.core.rv_pipeline import RVPipelineConfig


def scene_config_upgrader(cfg_dict: dict, version: int) -> dict:
    if version == 4:
        try:
            # removed in version 5
            if cfg_dict.get('aoi_geometries') is not None:
                raise ConfigError(
                    'SceneConfig.aoi_geometries is deprecated. '
                    'To use this config again, manually edit it to use '
                    'SceneConfig.aoi_uris instead.')
            del cfg_dict['aoi_geometries']
        except KeyError:
            pass
    return cfg_dict


[docs]@register_config('scene', upgrader=scene_config_upgrader) class SceneConfig(Config): """Configure a :class:`.Scene` comprising raster data & labels for an AOI. """ id: str raster_source: RasterSourceConfig label_source: Optional[LabelSourceConfig] = None label_store: Optional[LabelStoreConfig] = None aoi_uris: Optional[List[str]] = Field( None, description='List of URIs of GeoJSON files that define the AOIs for ' 'the scene. Each polygon defines an AOI which is a piece of the scene ' 'that is assumed to be fully labeled and usable for training or ' 'validation. The AOIs are assumed to be in EPSG:4326 coordinates.')
[docs] def build(self, class_config: 'ClassConfig', tmp_dir: Optional[str] = None, use_transformers: bool = True) -> Scene: raster_source = self.raster_source.build( tmp_dir, use_transformers=use_transformers) crs_transformer = raster_source.crs_transformer bbox = raster_source.bbox label_source = None label_store = None if self.label_source is not None: label_source = self.label_source.build( class_config=class_config, crs_transformer=crs_transformer, bbox=bbox, tmp_dir=tmp_dir) if self.label_store is not None: label_store = self.label_store.build( class_config=class_config, crs_transformer=crs_transformer, bbox=bbox, tmp_dir=tmp_dir) aoi_polygons = [] if self.aoi_uris is not None: aoi_polygons += get_polygons_from_uris(self.aoi_uris, crs_transformer) return Scene( self.id, raster_source, label_source=label_source, label_store=label_store, aoi_polygons=aoi_polygons)
[docs] def update(self, pipeline: Optional['RVPipelineConfig'] = None) -> None: super().update() self.raster_source.update(pipeline=pipeline, scene=self) if self.label_source is not None: self.label_source.update(pipeline=pipeline, scene=self) if self.label_store is None and pipeline is not None: self.label_store = pipeline.get_default_label_store(scene=self) if self.label_store is not None: self.label_store.update(pipeline=pipeline, scene=self)