SemanticSegmentationChipOptions#

Note

All Configs are derived from rastervision.pipeline.config.Config, which itself is a pydantic Model.

pydantic model SemanticSegmentationChipOptions[source]#

Chipping options for semantic segmentation.

Show JSON schema
{
   "title": "SemanticSegmentationChipOptions",
   "description": "Chipping options for semantic segmentation.",
   "type": "object",
   "properties": {
      "sampling": {
         "title": "Sampling",
         "description": "Window sampling config.",
         "anyOf": [
            {
               "$ref": "#/definitions/WindowSamplingConfig"
            },
            {
               "type": "object",
               "additionalProperties": {
                  "$ref": "#/definitions/WindowSamplingConfig"
               }
            }
         ]
      },
      "nodata_threshold": {
         "title": "Nodata Threshold",
         "description": "Discard chips where the proportion of NODATA values is greater than or equal to this value. Might result in false positives if there are many legitimate black pixels in the chip. Use with caution. If 1.0, only chips that are fully NODATA will be discarded. Defaults to 1.0.",
         "default": 1.0,
         "minimum": 0,
         "maximum": 1,
         "type": "number"
      },
      "type_hint": {
         "title": "Type Hint",
         "default": "semantic_segmentation_chip_options",
         "enum": [
            "semantic_segmentation_chip_options"
         ],
         "type": "string"
      },
      "target_class_ids": {
         "title": "Target Class Ids",
         "description": "List of class ids considered as targets (ie. those to prioritize when creating chips) which is only used in conjunction with the target_count_threshold and negative_survival_probability options. Applies to the random_sample window method.",
         "type": "array",
         "items": {
            "type": "integer"
         }
      },
      "negative_survival_prob": {
         "title": "Negative Survival Prob",
         "description": "Probability of keeping a negative chip.",
         "default": 1.0,
         "type": "number"
      },
      "target_count_threshold": {
         "title": "Target Count Threshold",
         "description": "Minimum number of pixels covering target_classes that a chip must have. Applies to the random_sample window method.",
         "default": 1000,
         "type": "integer"
      }
   },
   "required": [
      "sampling"
   ],
   "additionalProperties": false,
   "definitions": {
      "WindowSamplingMethod": {
         "title": "WindowSamplingMethod",
         "description": "Enum for window sampling methods.\n\nAttributes:\n    sliding: Sliding windows.\n    random: Randomly sampled windows.",
         "enum": [
            "sliding",
            "random"
         ]
      },
      "WindowSamplingConfig": {
         "title": "WindowSamplingConfig",
         "description": "Configure the sampling of chip windows.",
         "type": "object",
         "properties": {
            "method": {
               "default": "sliding",
               "allOf": [
                  {
                     "$ref": "#/definitions/WindowSamplingMethod"
                  }
               ]
            },
            "size": {
               "title": "Size",
               "description": "If method = sliding, this is the size of sliding window. If method = random, this is the size that all the windows are resized to before they are returned. If method = random and neither size_lims nor h_lims and w_lims have been specified, then size_lims is set to (size, size + 1).",
               "anyOf": [
                  {
                     "type": "integer",
                     "exclusiveMinimum": 0
                  },
                  {
                     "type": "array",
                     "minItems": 2,
                     "maxItems": 2,
                     "items": [
                        {
                           "type": "integer",
                           "exclusiveMinimum": 0
                        },
                        {
                           "type": "integer",
                           "exclusiveMinimum": 0
                        }
                     ]
                  }
               ]
            },
            "stride": {
               "title": "Stride",
               "description": "Stride of sliding window. Only used if method = sliding.",
               "anyOf": [
                  {
                     "type": "integer",
                     "exclusiveMinimum": 0
                  },
                  {
                     "type": "array",
                     "minItems": 2,
                     "maxItems": 2,
                     "items": [
                        {
                           "type": "integer",
                           "exclusiveMinimum": 0
                        },
                        {
                           "type": "integer",
                           "exclusiveMinimum": 0
                        }
                     ]
                  }
               ]
            },
            "padding": {
               "title": "Padding",
               "description": "How many pixels are windows allowed to overflow the edges of the raster source.",
               "anyOf": [
                  {
                     "type": "integer",
                     "minimum": 0
                  },
                  {
                     "type": "array",
                     "minItems": 2,
                     "maxItems": 2,
                     "items": [
                        {
                           "type": "integer",
                           "minimum": 0
                        },
                        {
                           "type": "integer",
                           "minimum": 0
                        }
                     ]
                  }
               ]
            },
            "pad_direction": {
               "title": "Pad Direction",
               "description": "If \"end\", only pad ymax and xmax (bottom and right). If \"start\", only pad ymin and xmin (top and left). If \"both\", pad all sides. Has no effect if paddiong is zero. Defaults to \"end\".",
               "default": "end",
               "enum": [
                  "both",
                  "start",
                  "end"
               ],
               "type": "string"
            },
            "size_lims": {
               "title": "Size Lims",
               "description": "[min, max) interval from which window sizes will be uniformly randomly sampled. The upper limit is exclusive. To fix the size to a constant value, use size_lims = (sz, sz + 1). Only used if method = random. Specify either size_lims or h_lims and w_lims, but not both. If neither size_lims nor h_lims and w_lims have been specified, then this will be set to (size, size + 1).",
               "type": "array",
               "minItems": 2,
               "maxItems": 2,
               "items": [
                  {
                     "type": "integer",
                     "exclusiveMinimum": 0
                  },
                  {
                     "type": "integer",
                     "exclusiveMinimum": 0
                  }
               ]
            },
            "h_lims": {
               "title": "H Lims",
               "description": "[min, max] interval from which window heights will be uniformly randomly sampled. Only used if method = random.",
               "type": "array",
               "minItems": 2,
               "maxItems": 2,
               "items": [
                  {
                     "type": "integer",
                     "exclusiveMinimum": 0
                  },
                  {
                     "type": "integer",
                     "exclusiveMinimum": 0
                  }
               ]
            },
            "w_lims": {
               "title": "W Lims",
               "description": "[min, max] interval from which window widths will be uniformly randomly sampled. Only used if method = random.",
               "type": "array",
               "minItems": 2,
               "maxItems": 2,
               "items": [
                  {
                     "type": "integer",
                     "exclusiveMinimum": 0
                  },
                  {
                     "type": "integer",
                     "exclusiveMinimum": 0
                  }
               ]
            },
            "max_windows": {
               "title": "Max Windows",
               "description": "Max number of windows to sample. Only used if method = random.",
               "default": 10000,
               "minimum": 0,
               "type": "integer"
            },
            "max_sample_attempts": {
               "title": "Max Sample Attempts",
               "description": "Max attempts when trying to find a window within the AOI of a scene. Only used if method = random and the scene has aoi_polygons specified.",
               "default": 100,
               "exclusiveMinimum": 0,
               "type": "integer"
            },
            "efficient_aoi_sampling": {
               "title": "Efficient Aoi Sampling",
               "description": "If the scene has AOIs, sampling windows at random anywhere in the extent and then checking if they fall within any of the AOIs can be very inefficient. This flag enables the use of an alternate algorithm that only samples window locations inside the AOIs. Only used if method = random and the scene has aoi_polygons specified. Defaults to True",
               "default": true,
               "type": "boolean"
            },
            "within_aoi": {
               "title": "Within Aoi",
               "description": "If True and if the scene has an AOI, only sample windows that lie fully within the AOI. If False, windows only partially intersecting the AOI will also be allowed.",
               "default": true,
               "type": "boolean"
            },
            "type_hint": {
               "title": "Type Hint",
               "default": "window_sampling",
               "enum": [
                  "window_sampling"
               ],
               "type": "string"
            }
         },
         "required": [
            "size"
         ],
         "additionalProperties": false
      }
   }
}

Config

  • extra: str = forbid

  • validate_assignment: bool = True

Fields
field negative_survival_prob: float = 1.0#

Probability of keeping a negative chip.

field nodata_threshold: Proportion = 1.0#

Discard chips where the proportion of NODATA values is greater than or equal to this value. Might result in false positives if there are many legitimate black pixels in the chip. Use with caution. If 1.0, only chips that are fully NODATA will be discarded. Defaults to 1.0.

Constraints

  • minimum = 0

  • maximum = 1

field sampling: Union[WindowSamplingConfig, Dict[str, WindowSamplingConfig]] [Required]#

Window sampling config.

field target_class_ids: Optional[List[int]] = None#

List of class ids considered as targets (ie. those to prioritize when creating chips) which is only used in conjunction with the target_count_threshold and negative_survival_probability options. Applies to the random_sample window method.

field target_count_threshold: int = 1000#

Minimum number of pixels covering target_classes that a chip must have. Applies to the random_sample window method.

field type_hint: Literal['semantic_segmentation_chip_options'] = 'semantic_segmentation_chip_options'#
build()#

Build an instance of the corresponding type of object using this config.

For example, BackendConfig will build a Backend object. The arguments to this method will vary depending on the type of Config.

classmethod deserialize(inp: str | dict | Config) → Self#

Deserialize Config from a JSON file or dict, upgrading if possible.

If inp is already a Config, it is returned as is.

Parameters

inp (str | dict | Config) – a URI to a JSON file or a dict.

Return type

Self

enough_target_pixels(label_arr: ndarray) → bool[source]#

Check if label raster has enough pixels of the target classes.

Parameters

label_arr (ndarray) – The label raster for a chip.

Returns

True (the window does contain interesting pixels) or False.

Return type

bool

classmethod from_dict(cfg_dict: dict) → Self#

Deserialize Config from a dict.

Parameters

cfg_dict (dict) – Dict to deserialize.

Return type

Self

classmethod from_file(uri: str) → Self#

Deserialize Config from a JSON file, upgrading if possible.

Parameters

uri (str) – URI to load from.

Return type

Self

get_chip_sz(scene_id: Optional[str] = None) → int#
Parameters

scene_id (Optional[str]) –

Return type

int

keep_chip(chip: ndarray, label: ndarray) → bool[source]#

Decide whether to keep or discard chip.

Parameters

  • chip (ndarray) – Chip raster.

  • label (ndarray) – Associated label.

Return type

bool

recursive_validate_config()#

Recursively validate hierarchies of Configs.

This uses reflection to call validate_config on a hierarchy of Configs using a depth-first pre-order traversal.

revalidate()#

Re-validate an instantiated Config.

Runs all Pydantic validators plus self.validate_config().

Adapted from: https://github.com/samuelcolvin/pydantic/issues/1864#issuecomment-679044432

to_file(uri: str, with_rv_metadata: bool = True) → None#

Save a Config to a JSON file, optionally with RV metadata.

Parameters

  • uri (str) – URI to save to.

  • with_rv_metadata (bool) – If True, inject Raster Vision metadata such as plugin_versions, so that the config can be upgraded when loaded.

Return type

None

update(*args, **kwargs)#

Update any fields before validation.

Subclasses should override this to provide complex default behavior, for example, setting default values as a function of the values of other fields. The arguments to this method will vary depending on the type of Config.

validate_config()#

Validate fields that should be checked after update is called.

This is to complement the builtin validation that Pydantic performs at the time of object construction.

validate_list(field: str, valid_options: List[str])#

Validate a list field.

Parameters

  • field (str) – name of field to validate

  • valid_options (List[str]) – values that field is allowed to take

Raises

ConfigError – if field is invalid