luxonis_ml.data.augmentations.albumentations_engine.AlbumentationConfigItem

class documentation

class AlbumentationConfigItem(ConfigItem):

Configuration item for AlbumentationsEngine.

Instance Variable	`apply_on_stages`	List of pipeline stages to apply this transformation on. Valid stages are `"train"`, `"val"`, and `"test"`. By default, transformations are applied only during the `"train"` stage.
Instance Variable	`name`	Name of the transformation. Must be either a valid name of an Albumentations transformation, or a name of a custom transformation registered in the `TRANSFORMATIONS` registry.
Instance Variable	`params`	Parameters for the transformation.
Instance Variable	`use_for_resizing`	Whether this transformation is eligible to be used for resizing.

Inherited from BaseModelExtraForbid (via ConfigItem):

Instance Variable model_config Pydantic model configuration with extra set to "forbid".

apply_on_stages: list[Literal['train', 'val', 'test']] = ¶

List of pipeline stages to apply this transformation on. Valid stages are "train", "val", and "test". By default, transformations are applied only during the "train" stage.

name = ¶

overrides luxonis_ml.typing.ConfigItem.name

Name of the transformation. Must be either a valid name of an Albumentations transformation, or a name of a custom transformation registered in the TRANSFORMATIONS registry.

params = ¶

overrides luxonis_ml.typing.ConfigItem.params

Parameters for the transformation.

use_for_resizing: bool = ¶

Whether this transformation is eligible to be used for resizing.