Source code for rastervision.core.rv_pipeline.chip_classification_config
from rastervision.pipeline.config import register_config
from rastervision.core.rv_pipeline import RVPipelineConfig
from rastervision.core.data.label_store import (
ChipClassificationGeoJSONStoreConfig)
from rastervision.core.evaluation import ChipClassificationEvaluatorConfig
[docs]@register_config('chip_classification')
class ChipClassificationConfig(RVPipelineConfig):
"""Configure a :class:`.ChipClassification` pipeline."""
[docs] def build(self, tmp_dir):
from rastervision.core.rv_pipeline.chip_classification import ChipClassification
return ChipClassification(self, tmp_dir)