Skip to content

[FEATURE] Simpler Augmentation Config #34

Description

@triwahyuu

Is your feature request related to a problem? Please describe.
current augmentation config could be complex, hard to read and understand, and seemingly redundant.

augmentations : [
  {
    module : albumentations,
    args : {
      transforms : [
        {compose : OneOf, args : {
          transforms : [{transform : RandomBrightnessContrast, args : {p : 0.5}},
                        {transform : RandomSnow, args : {p : 0.5}}
          ],
          p : 0.5}},
        {transform : HorizontalFlip, args : {p : 0.5}},
        {transform : RandomScale, args : {scale_limit : 0.3, p : 0.5,}}
      ],
      bbox_params : {
        min_visibility : 0.0,
        min_area : 0.0
      },
      visual_debug : False
    }
  }
]

Describe the solution you'd like

  • treat compose and transform as the same (at least in config)
  • no need to specify 'transform' and 'compose' separately from 'args'

example:

augmentations : [
  {
    albumentations: {
      transforms : [
        {
          OneOf: {
            transforms : [
              {RandomBrightnessContrast: {p : 0.5}},
              {RandomSnow: {p : 0.5}}
            ],
            p : 0.5
          }
        },
        {HorizontalFlip: {p : 0.5}},
        {RandomScale: {scale_limit : 0.3, p : 0.5,}}
      ],
      bbox_params : {
        min_visibility : 0.0,
        min_area : 0.0
      },
      visual_debug : False
    }
  }
]

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Fields

    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions