nodefluxio / vortex

A Deep Learning Model Development Framework for Computer Vision
27 stars 6 forks source link

[FEATURE] Simpler Augmentation Config #34

Open triwahyuu opened 4 years ago

triwahyuu commented 4 years ago

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

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
    }
  }
]
alphinside commented 4 years ago

good idea