def color_jitter_list()

in lib/datasets/image_processor.py [0:0]

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

def color_jitter_list(images, img_brightness=0, img_contrast=0, img_saturation=0):

    jitter = []
    if img_brightness != 0:
        jitter.append('brightness')
    if img_contrast != 0:
        jitter.append('contrast')
    if img_saturation != 0:
        jitter.append('saturation')

    if len(jitter) > 0:
        order = np.random.permutation(np.arange(len(jitter)))
        for idx in range(0, len(jitter)):
            if jitter[order[idx]] == 'brightness':
                images = brightness_list(img_brightness, images)
            elif jitter[order[idx]] == 'contrast':
                images = contrast_list(img_contrast, images)
            elif jitter[order[idx]] == 'saturation':
                images = saturation_list(img_saturation, images)
    return images