ObjectDetectionChipOptions#

Note

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

pydantic model ObjectDetectionChipOptions[source]#

Show JSON schema
{
   "title": "ObjectDetectionChipOptions",
   "description": "Configure the sampling and filtering of chips.",
   "type": "object",
   "properties": {
      "sampling": {
         "title": "Sampling",
         "description": "Window sampling config.",
         "allOf": [
            {
               "$ref": "#/definitions/ObjectDetectionWindowSamplingConfig"
            }
         ]
      },
      "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": "object_detection_chip_options",
         "enum": [
            "object_detection_chip_options"
         ],
         "type": "string"
      }
   },
   "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"
         ]
      },
      "ObjectDetectionWindowSamplingConfig": {
         "title": "ObjectDetectionWindowSamplingConfig",
         "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"
            },
            "type_hint": {
               "title": "Type Hint",
               "default": "object_detection_window_sampling",
               "enum": [
                  "object_detection_window_sampling"
               ],
               "type": "string"
            },
            "ioa_thresh": {
               "title": "Ioa Thresh",
               "description": "When a box is partially outside of a training chip, it is not clear if (a clipped version) of the box should be included in the chip. If the IOA (intersection over area) of the box with the chip is greater than ioa_thresh, it is included in the chip. Defaults to 0.8.",
               "default": 0.8,
               "type": "number"
            },
            "clip": {
               "title": "Clip",
               "description": "Clip bounding boxes to window limits when retrieving labels for a window.",
               "default": false,
               "type": "boolean"
            },
            "neg_ratio": {
               "title": "Neg Ratio",
               "description": "The ratio of negative chips (those containing no bounding boxes) to positive chips. This can be useful if the statistics of the background is different in positive chips. For example, in car detection, the positive chips will always contain roads, but no examples of rooftops since cars tend to not be near rooftops. Defaults to None.",
               "type": "number"
            },
            "neg_ioa_thresh": {
               "title": "Neg Ioa Thresh",
               "description": "A window will be considered negative if its max IoA with any bounding box is less than this threshold. Defaults to 0.2.",
               "default": 0.2,
               "type": "number"
            }
         },
         "required": [
            "size"
         ],
         "additionalProperties": false
      }
   }
}

Config
  • extra: str = forbid

  • validate_assignment: bool = True

Fields
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: ObjectDetectionWindowSamplingConfig [Required]#

Window sampling config.

field type_hint: Literal['object_detection_chip_options'] = 'object_detection_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 from_file(uri: str) Config#

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

Parameters

uri (str) – URI to load from.

Return type

Config

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

scene_id (Optional[str]) –

Return type

int

keep_chip(chip: ndarray, label: Any) bool#

Decide whether to keep or discard chip.

Parameters
  • chip (ndarray) – Chip raster.

  • label (Any) – 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