def build_augmenters()

in broadcast-monitoring/scripts/generate-logo-images.py [0:0]

• 25 lines of code
• 9 McCabe index (conditional complexity)

def build_augmenters(args, augmenters=None):
    ''' Construct a list of augmenters which rely on the arguments handled by the parent parser'''

    if augmenters is None:
        augmenters = []

    opacity_range = args.opacity
    if opacity_range != (-1.0, -1.0):
        # set the alpha channel to a random value
        if opacity_range == (1.0, 1.0):
            opacity_range = (1.0)
        augmenters.insert(0, iaa.Lambda(func_images=set_alpha_wrapper(opacity_range)))

    scale = args.scale
    if scale != (1.0, 1.0):
        augmenters.append(iaa.Affine(scale=scale))    # scale the image between 20% to 125% of its original size

    rotate = args.rotate
    if rotate != (0.0, 0.0):
        augmenters.append(iaa.Rotate(rotate))    # rotate the image somewhere in the range of -10.0 - 10.0 deg

    shear = args.shear
    if shear != (0.0, 0.0):
        augmenters.append(iaa.ShearX(shear))    # rotate the image somewhere in the range of -10.0 - 10.0 deg

    noise = args.noise
    if noise != (0, 0):
        augmenters.append(iaa.AdditiveGaussianNoise(scale=noise))

    grayscale = args.grayscale
    if grayscale != (0.0, 1.0):
        augmenters.append(iaa.WithChannels([0, 1, 2], iaa.Grayscale(alpha=(0.0, 1.0))))

    return augmenters